RomanNum3ral
/
ComfyUI
mirror of https://github.com/Comfy-Org/ComfyUI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Support MiniMax-H3 loras trained with DiffSynth-Studio.

This commit is contained in:
yjy415 2026-08-16 11:40:15 +08:00
parent 5ba6947908
commit 301f02d019
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy/lora_convert.py
View File

@ -32,6 +32,12 @@ def convert_uso_lora(sd):
sd_out[k_to] = tensor
return sd_out
def convert_lora_minimax_h3(sd):
sd_out = {}
for k in sd:
sd_out["diffusion_model.{}".format(k)] = sd[k]
return sd_out
def convert_lora(sd):
if "img_in.lora_A.weight" in sd and "single_blocks.0.norm.key_norm.scale" in sd:
@ -40,4 +46,6 @@ def convert_lora(sd):
return convert_lora_wan_fun(sd)
if "single_blocks.37.processor.qkv_lora.up.weight" in sd and "double_blocks.18.processor.qkv_lora2.up.weight" in sd:
return convert_uso_lora(sd)
if "blocks.0.attn.qkv_proj.lora_A.default.weight" in sd:
return convert_lora_minimax_h3(sd)
return sd