Skip to content

GitLab

Commit 9424522e authored by comfyanonymous's avatar comfyanonymous
Browse files

Reuse code.

parent 73ce1780
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 6 deletions
comfy/model_base.py
View file @ 9424522e
...@@ -573,13 +573,8 @@ class SD3(BaseModel): ...@@ -573,13 +573,8 @@ class SD3(BaseModel):
return kwargs["pooled_output"] return kwargs["pooled_output"]
def extra_conds(self, **kwargs): def extra_conds(self, **kwargs):
out = {} out = super().extra_conds(**kwargs)
adm = self.encode_adm(**kwargs)
if adm is not None:
out['y'] = comfy.conds.CONDRegular(adm)
cross_attn = kwargs.get("cross_attn", None) cross_attn = kwargs.get("cross_attn", None)
if cross_attn is not None: if cross_attn is not None:
out['c_crossattn'] = comfy.conds.CONDRegular(cross_attn) out['c_crossattn'] = comfy.conds.CONDRegular(cross_attn)
return out return out
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