Skip to content

Commit 3a56201

Browse files
Allow flux conditioning without a pooled output. (#13198)
1 parent 6a2cdb8 commit 3a56201

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

comfy/model_base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -890,7 +890,7 @@ def concat_cond(self, **kwargs):
890890
return torch.cat((image, mask), dim=1)
891891

892892
def encode_adm(self, **kwargs):
893-
return kwargs["pooled_output"]
893+
return kwargs.get("pooled_output", None)
894894

895895
def extra_conds(self, **kwargs):
896896
out = super().extra_conds(**kwargs)

0 commit comments

Comments
 (0)