lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-27 01:38:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix lora not loaded for text encoder #142

Browse Source
This commit is contained in:
lllyasviel
2024-02-09 13:50:50 -08:00
parent 200f2b69ed
commit 8059533eaf
3 changed files with 14 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
modules/sd_models_xl.py
View File

@@ -9,10 +9,11 @@ from modules import devices, shared, prompt_parser
from modules import torch_utils
import ldm_patched.modules.model_management as model_management
from modules_forge.forge_clip import move_clip_to_gpu
def get_learned_conditioning(self: sgm.models.diffusion.DiffusionEngine, batch: prompt_parser.SdConditioning | list[str]):
model_management.load_model_gpu(self.forge_objects.clip.patcher)
move_clip_to_gpu()
for embedder in self.conditioner.embedders:
embedder.ucg_rate = 0.0