Let conditioning specify custom concat conds.

MoonRide303 · Mar 2, 2024 · 51df846 · 51df846
1 parent 9f71e4b
commit 51df846
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/comfy/model_base.py b/comfy/model_base.py
@@ -167,6 +167,10 @@ def blank_inpaint_image_like(latent_image):
         if cross_attn_cnet is not None:
             out['crossattn_controlnet'] = comfy.conds.CONDCrossAttn(cross_attn_cnet)
 
+        c_concat = kwargs.get("noise_concat", None)
+        if c_concat is not None:
+            out['c_concat'] = comfy.conds.CONDNoiseShape(data)
+
         return out
 
     def load_model_weights(self, sd, unet_prefix=""):