lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-02 19:49:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update forge_clip.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 06:55:26 -08:00
parent ba108837f8
commit 6d429e9498
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modules_forge/forge_clip.py
View File

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