From 39866091882903f9b8687a9769ecb8fcffccb279 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 10:27:53 -0800 Subject: [PATCH] Update forge_clip.py --- modules_forge/forge_clip.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/modules_forge/forge_clip.py b/modules_forge/forge_clip.py index 9084c16e..0e6e3539 100644 --- a/modules_forge/forge_clip.py +++ b/modules_forge/forge_clip.py @@ -1,5 +1,3 @@ -import torch - from modules.sd_hijack_clip import FrozenCLIPEmbedderWithCustomWords from ldm_patched.modules import model_management @@ -46,6 +44,10 @@ class CLIP_SD_XL_G(FrozenCLIPEmbedderWithCustomWords): self.wrapped.layer = "hidden" self.wrapped.layer_idx = -2 + self.id_start = 49406 + self.id_end = 49407 + self.id_pad = 0 + def encode_with_transformers(self, tokens): outputs = self.wrapped.transformer(tokens, output_hidden_states=self.wrapped.layer == "hidden")