diff --git a/app/user_manager.py b/app/user_manager.py index bb6217995..e863b93dd 100644 --- a/app/user_manager.py +++ b/app/user_manager.py @@ -36,7 +36,7 @@ class UserManager(): self.settings = AppSettings(self) if not os.path.exists(user_directory): - os.mkdir(user_directory) + os.makedirs(user_directory, exist_ok=True) if not args.multi_user: print("****** User settings have been changed to be stored on the server instead of browser storage. ******") print("****** For multi-user setups add the --multi-user CLI argument to enable multiple user profiles. ******") diff --git a/comfy/model_base.py b/comfy/model_base.py index 3c385cda9..c305014a4 100644 --- a/comfy/model_base.py +++ b/comfy/model_base.py @@ -712,7 +712,13 @@ class Flux(BaseModel): super().__init__(model_config, model_type, device=device, unet_model=comfy.ldm.flux.model.Flux) def concat_cond(self, **kwargs): - num_channels = self.diffusion_model.img_in.weight.shape[1] // (self.diffusion_model.patch_size * self.diffusion_model.patch_size) + try: + #Handle Flux control loras dynamically changing the img_in weight. + num_channels = self.diffusion_model.img_in.weight.shape[1] // (self.diffusion_model.patch_size * self.diffusion_model.patch_size) + except: + #Some cases like tensorrt might not have the weights accessible + num_channels = self.model_config.unet_config["in_channels"] + out_channels = self.model_config.unet_config["out_channels"] if num_channels <= out_channels: