OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-03-05 05:24:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

check lora_unet prefix to support Black Forest Labs's lora

Browse Source
This commit is contained in:
Won-Kyu Park 2024-09-19 09:29:46 +09:00
parent 6675d1f090
commit 1f779226f0
No known key found for this signature in database
GPG Key ID: 53AA79C8C9535D15
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -183,8 +183,12 @@ def load_network(name, network_on_disk):
for key_network, weight in sd.items(): for key_network, weight in sd.items():
if diffusers_weight_map: if diffusers_weight_map:
key_network_without_network_parts, network_name, network_weight = key_network.rsplit(".", 2) if key_network.startswith("lora_unet"):
network_part = network_name + '.' + network_weight key_network_without_network_parts, _, network_part = key_network.partition(".")
key_network_without_network_parts = key_network_without_network_parts.replace("lora_unet", "diffusion_model")
else:
key_network_without_network_parts, network_name, network_weight = key_network.rsplit(".", 2)
network_part = network_name + '.' + network_weight
else: else:
key_network_without_network_parts, _, network_part = key_network.partition(".") key_network_without_network_parts, _, network_part = key_network.partition(".")