lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-24 08:43:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

i

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 12:19:50 -08:00
parent 668cef32e4
commit 1e7becbeb6
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modules/sd_models_xl.py
View File

@@ -12,7 +12,7 @@ import ldm_patched.modules.model_management as model_management
def get_learned_conditioning(self: sgm.models.diffusion.DiffusionEngine, batch: prompt_parser.SdConditioning | list[str]):
model_management.load_model_gpu(self.clip_patcher)
model_management.load_model_gpu(self.forge_objects.clip.patcher)
for embedder in self.conditioner.embedders:
embedder.ucg_rate = 0.0

4
modules_forge/forge_clip.py
View File

@@ -5,7 +5,7 @@ from modules.shared import opts
class CLIP_SD_15_L(FrozenCLIPEmbedderWithCustomWords):
def encode_with_transformers(self, tokens):
model_management.load_models_gpu([self.patcher.patcher])
model_management.load_model_gpu(self.forge_objects.clip.patcher)
outputs = self.wrapped.transformer(input_ids=tokens, output_hidden_states=-opts.CLIP_stop_at_last_layers)
if opts.CLIP_stop_at_last_layers > 1:
@@ -30,7 +30,7 @@ class CLIP_SD_21_H(FrozenCLIPEmbedderWithCustomWords):
self.id_pad = 0
def encode_with_transformers(self, tokens):
model_management.load_models_gpu([self.patcher.patcher])
model_management.load_model_gpu(self.forge_objects.clip.patcher)
outputs = self.wrapped.transformer(tokens, output_hidden_states=self.wrapped.layer == "hidden")
if self.wrapped.layer == "last":