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

Update networks.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 14:55:10 -08:00
parent 8954f95c82
commit 71be477d2f
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
extensions-builtin/Lora/networks.py
View File

@@ -56,14 +56,15 @@ def load_networks(names, te_multipliers=None, unet_multipliers=None, dyn_dims=No
return
current_sd.current_lora_hash = compiled_lora_targets_hash
current_sd.forge_objects.unet = current_sd.forge_objects.unet_original
current_sd.forge_objects.clip = current_sd.forge_objects.clip_original
current_sd.forge_objects.unet = current_sd.forge_objects_original.unet
current_sd.forge_objects.clip = current_sd.forge_objects_original.clip
for filename, strength_model, strength_clip in compiled_lora_targets:
lora_sd = load_lora_state_dict(filename)
current_sd.forge_objects.unet, current_sd.forge_objects.clip = load_lora_for_models(
current_sd.forge_objects.unet, current_sd.forge_objects.clip, lora_sd, strength_model, strength_clip)
current_sd.forge_objects_after_applying_lora = current_sd.forge_objects.shallow_copy()
return