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

2026-03-27 17:36:26 -07:00
parent 6a2cdb817d
commit 3a56201da5
1 changed files with 1 additions and 1 deletions
@@ -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)