From 6d429e94984d2f9b0c9b289cc67b2c2af526972a Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 06:55:26 -0800 Subject: [PATCH] Update forge_clip.py --- modules_forge/forge_clip.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/modules_forge/forge_clip.py b/modules_forge/forge_clip.py index 106cbfe0..db814e31 100644 --- a/modules_forge/forge_clip.py +++ b/modules_forge/forge_clip.py @@ -4,6 +4,7 @@ from ldm_patched.modules import model_management class CLIP_SD_15_L(FrozenCLIPEmbedderWithCustomWords): def encode_with_transformers(self, tokens): + model_management.load_models_gpu([self.patcher.patcher]) return super().encode_with_transformers(tokens) @@ -16,6 +17,7 @@ class CLIP_SD_21_G(FrozenCLIPEmbedderWithCustomWords): self.wrapped.layer_idx = -2 def encode_with_transformers(self, tokens): + model_management.load_models_gpu([self.patcher.patcher]) return super().encode_with_transformers(tokens)