zlei6/ComfyUI
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI.git synced 2026-08-12 21:04:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add LoRA key mapping for LTXV/LTXAV models (#14349)

Browse Source
This commit is contained in:
kelseyee
2026-06-09 21:57:58 +08:00
committed by GitHub
parent 1639dc7a70
commit 07c53f8f0f
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/lora.py
View File

@@ -357,6 +357,12 @@ def model_lora_keys_unet(model, key_map={}):
key_lora = k[len("diffusion_model."):-len(".weight")]
key_map["transformer.{}".format(key_lora)] = k
if isinstance(model, (comfy.model_base.LTXV, comfy.model_base.LTXAV)):
for k in sdk:
if k.startswith("diffusion_model.") and k.endswith(".weight"):
key_lora = k[len("diffusion_model."):-len(".weight")]
key_map["{}".format(key_lora)] = k
return key_map