wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-25 09:38:10 +08:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

Merge c4f414081c into 6665515349

Browse Source
This commit is contained in:
Jinyan Ye
2026-07-19 19:40:23 -07:00
committed by GitHub
parent 6665515349 c4f414081c
commit c86d91bee1
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
comfy/lora.py
+7
View File
@@ -304,6 +304,13 @@ def model_lora_keys_unet(model, key_map={}):
key_map["lycoris_{}".format(key_lora.replace(".", "_"))] = k #SimpleTuner lycoris format
key_map["transformer.{}".format(key_lora)] = k #SimpleTuner regular format
if isinstance(model, comfy.model_base.HiDreamO1):
for k in sdk:
if k.startswith("diffusion_model."):
if k.endswith(".weight"):
key_lora = k[len("diffusion_model."):-len(".weight")]
key_map["model.{}".format(key_lora)] = k
if isinstance(model, comfy.model_base.ACEStep):
for k in sdk:
if k.startswith("diffusion_model.") and k.endswith(".weight"): #Official ACE step lora format