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

Update forge_loader.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 16:22:29 -08:00
parent abf33dbb43
commit 48e9804ffb
1 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
modules_forge/forge_loader.py
View File

@@ -241,8 +241,21 @@ def load_model_for_a1111(timer, checkpoint_info=None, state_dict=None):
sd_model.decode_first_stage = patched_decode_first_stage
sd_model.encode_first_stage = patched_encode_first_stage
patch_unet_forward(sd_model)
sd_model.clip = sd_model.cond_stage_model
timer.record("forge finalize")
sd_model.current_lora_hash = str([])
return sd_model
def patch_unet_forward(sd_model):
original_forward = sd_model.model.diffusion_model.forward
def forge_unet_forward(self, *args, **kwargs):
return original_forward(self, *args, **kwargs)
sd_model.model.diffusion_model.forward = forge_unet_forward
return