Update forge_clip.py

This commit is contained in:
lllyasviel
2024-01-25 06:23:30 -08:00
parent 035aca8ad9
commit 1d8275e982

View File

@@ -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