From e8d0736cb349d9bc34eac41b39f4d3df8104dd6a Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 06:51:38 -0800 Subject: [PATCH] Update forge_clip.py --- modules_forge/forge_clip.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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):