Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/ComfyUI.git" did not exist on "f6a203951f061759a3d6a9847b865dd9c7d77a38"
Commit 51df8465 authored by comfyanonymous's avatar comfyanonymous
Browse files

Let conditioning specify custom concat conds.

parent 9f71e4b6
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
comfy/model_base.py
View file @ 51df8465
...@@ -167,6 +167,10 @@ class BaseModel(torch.nn.Module): ...@@ -167,6 +167,10 @@ class BaseModel(torch.nn.Module):
if cross_attn_cnet is not None: if cross_attn_cnet is not None:
out['crossattn_controlnet'] = comfy.conds.CONDCrossAttn(cross_attn_cnet) 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 return out
def load_model_weights(self, sd, unet_prefix=""): def load_model_weights(self, sd, unet_prefix=""):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment