soaska/my-sd
soaska/my-sd
1
0
Fork 0
Code Issues Pull Requests Actions Packages Activity

Update forge_loader.py

Browse Source
This commit is contained in:
lllyasviel 2024-01-25 16:23:04 -08:00
parent 48e9804ffb
commit 79c7f0f513
1 changed files with 5 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
modules_forge/forge_loader.py
View File

@ -241,16 +241,6 @@ 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):
@ -258,4 +248,8 @@ def patch_unet_forward(sd_model):
sd_model.model.diffusion_model.forward = forge_unet_forward
return
sd_model.clip = sd_model.cond_stage_model
timer.record("forge finalize")
sd_model.current_lora_hash = str([])
return sd_model