298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-19 02:43:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Reuse code.

Browse Source
This commit is contained in:
comfyanonymous 2024-06-11 07:20:26 -04:00
parent 73ce178021
commit 9424522ead
1 changed files with 1 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
comfy/model_base.py
View File

@ -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