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")