Properly load the newbie diffusion model. (#11172)

There is still one of the text encoders missing and I didn't actually test it.
This commit is contained in:
comfyanonymous
2025-12-07 04:44:55 -08:00
committed by GitHub
parent 329480da5a
commit 56fa7dbe38
3 changed files with 42 additions and 0 deletions

View File

@@ -1110,6 +1110,10 @@ class Lumina2(BaseModel):
if 'num_tokens' not in out:
out['num_tokens'] = comfy.conds.CONDConstant(cross_attn.shape[1])
clip_text_pooled = kwargs["pooled_output"] # Newbie
if clip_text_pooled is not None:
out['clip_text_pooled'] = comfy.conds.CONDRegular(clip_text_pooled)
return out
class WAN21(BaseModel):