Add a way to set different conditioning for the controlnet.

2024-02-09 14:13:31 -05:00
parent fd73b5ee3a
commit 25a4805e51
2 changed files with 5 additions and 1 deletions
--- a/comfy/model_base.py
+++ b/comfy/model_base.py
@@ -153,6 +153,10 @@ class BaseModel(torch.nn.Module):
        if cross_attn is not None:
            out['c_crossattn'] = comfy.conds.CONDCrossAttn(cross_attn)

+        cross_attn_cnet = kwargs.get("cross_attn_controlnet", None)
+        if cross_attn_cnet is not None:
+            out['crossattn_controlnet'] = comfy.conds.CONDCrossAttn(cross_attn_cnet)
+
        return out

    def load_model_weights(self, sd, unet_prefix=""):