Allow flux conditioning without a pooled output. (#13198)

This commit is contained in:
comfyanonymous 2026-03-27 17:36:26 -07:00 committed by GitHub
parent 6a2cdb817d
commit 3a56201da5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -890,7 +890,7 @@ class Flux(BaseModel):
return torch.cat((image, mask), dim=1)
def encode_adm(self, **kwargs):
return kwargs["pooled_output"]
return kwargs.get("pooled_output", None)
def extra_conds(self, **kwargs):
out = super().extra_conds(**kwargs)