lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-09 15:09:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update networks.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 12:08:06 -08:00
parent e747c8c564
commit a1444c92e3
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -135,6 +135,8 @@ 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.unet_patcher, current_sd.clip_patcher = \
current_sd.unet_patcher_original, current_sd.clip_patcher_original
for filename, strength_model, strength_clip in compiled_lora_targets:
if filename in lora_state_dict_cache:
@@ -142,11 +144,12 @@ def load_networks(names, te_multipliers=None, unet_multipliers=None, dyn_dims=No
else:
if len(lora_state_dict_cache) > lora_state_dict_cache_max_length:
lora_state_dict_cache = {}
lora_sd = load_torch_file(filename, safe_load=True)
lora_state_dict_cache[filename] = lora_sd
a = 0
current_sd.unet_patcher, current_sd.clip_patcher = load_lora_for_models(
current_sd.unet_patcher, current_sd.clip_patcher, lora_sd, strength_model, strength_clip)
return