298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-25 15:55:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support loras based on the stability unet implementation.

Browse Source
This commit is contained in:
comfyanonymous 2023-06-26 02:56:11 -04:00
parent c71a7e6b20
commit b72a7a835a
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/sd.py
View File

@ -285,6 +285,11 @@ def model_lora_keys(model, key_map={}):
if key_in:
counter += 1
for k in sdk:
if k.startswith("diffusion_model.") and k.endswith(".weight"):
key_lora = k[len("diffusion_model."):-len(".weight")].replace(".", "_")
key_map["lora_unet_{}".format(key_lora)] = k
return key_map