mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-04-03 07:23:30 +08:00
Allow flux conditioning without a pooled output. (#13198)
This commit is contained in:
parent
6a2cdb817d
commit
3a56201da5
@ -890,7 +890,7 @@ class Flux(BaseModel):
|
|||||||
return torch.cat((image, mask), dim=1)
|
return torch.cat((image, mask), dim=1)
|
||||||
|
|
||||||
def encode_adm(self, **kwargs):
|
def encode_adm(self, **kwargs):
|
||||||
return kwargs["pooled_output"]
|
return kwargs.get("pooled_output", None)
|
||||||
|
|
||||||
def extra_conds(self, **kwargs):
|
def extra_conds(self, **kwargs):
|
||||||
out = super().extra_conds(**kwargs)
|
out = super().extra_conds(**kwargs)
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user