From 0ed2099601934a951697f10bcc52ca8be9118cde Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 06:16:31 -0800 Subject: [PATCH] Update forge_clip.py --- modules_forge/forge_clip.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/modules_forge/forge_clip.py b/modules_forge/forge_clip.py index e83209f6..c1b36ad6 100644 --- a/modules_forge/forge_clip.py +++ b/modules_forge/forge_clip.py @@ -28,6 +28,10 @@ class CLIP_SD_XL_G(FrozenCLIPEmbedderWithCustomWords): def __init__(self, wrapped, hijack): super().__init__(wrapped, hijack) + if self.wrapped.layer == "penultimate": + self.wrapped.layer = "hidden" + self.wrapped.layer_idx = -2 + def encode_with_transformers(self, tokens): outputs = self.wrapped.transformer(input_ids=tokens, output_hidden_states=self.wrapped.layer == "hidden")