Update forge_clip.py

This commit is contained in:
lllyasviel 2024-01-25 10:38:36 -08:00
parent b6d11f0d0b
commit d2f5e8c353

View File

@ -16,6 +16,10 @@ class CLIP_SD_21_H(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):
model_management.load_models_gpu([self.patcher.patcher])
return super().encode_with_transformers(tokens)