Skip to content

GitLab

"...git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "0d7ee968077096cb010896e28061bc4bcb2eb40a"
Commit 51df8465 authored by comfyanonymous's avatar comfyanonymous
Browse files

Let conditioning specify custom concat conds.

parent 9f71e4b6
Hide 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