From 497db6212faea643d3aac1253ccfa23834109881 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Tue, 26 Nov 2024 17:51:40 -0500 Subject: [PATCH] Alternative fix for #5767 --- comfy/model_base.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/comfy/model_base.py b/comfy/model_base.py index 3c385cda..c305014a 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: