diff --git a/modules_forge/forge_clip.py b/modules_forge/forge_clip.py index 7079c4b9..0e718e22 100644 --- a/modules_forge/forge_clip.py +++ b/modules_forge/forge_clip.py @@ -2,7 +2,8 @@ from modules.sd_hijack_clip import FrozenCLIPEmbedderWithCustomWords class CLIP_SD_15_L(FrozenCLIPEmbedderWithCustomWords): - pass + def encode_with_transformers(self, tokens): + return super().encode_with_transformers(tokens) class CLIP_SD_21_G(FrozenCLIPEmbedderWithCustomWords): @@ -13,6 +14,9 @@ class CLIP_SD_21_G(FrozenCLIPEmbedderWithCustomWords): self.wrapped.layer = "hidden" self.wrapped.layer_idx = -2 + def encode_with_transformers(self, tokens): + return super().encode_with_transformers(tokens) + class CLIP_SD_XL_L(FrozenCLIPEmbedderWithCustomWords): def __init__(self, wrapped, hijack):