diff --git a/modules_forge/forge_clip.py b/modules_forge/forge_clip.py index 06ca4474..7079c4b9 100644 --- a/modules_forge/forge_clip.py +++ b/modules_forge/forge_clip.py @@ -45,4 +45,6 @@ class CLIP_SD_XL_G(FrozenCLIPEmbedderWithCustomWords): else: z = outputs.hidden_states[self.wrapped.layer_idx] + z.pooled = outputs.pooler_output + return z