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

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)