From 6dc590041f669111e8385a79a07a2c372e46a8c6 Mon Sep 17 00:00:00 2001
From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com>
Date: Fri, 27 Mar 2026 20:31:47 -0400
Subject: [PATCH] Allow flux conditioning without a pooled output.

---
 comfy/model_base.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/comfy/model_base.py b/comfy/model_base.py
index 70aff886e..94579fa3e 100644
--- a/comfy/model_base.py
+++ b/comfy/model_base.py
@@ -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)