From 64e9c1edb4dd964042bf5860b85bc925582ffead Mon Sep 17 00:00:00 2001 From: rickard Date: Wed, 5 Feb 2025 07:46:47 +0100 Subject: [PATCH] also load hunyuan loras --- comfy/lora.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/comfy/lora.py b/comfy/lora.py index e3ddb904..44a52676 100644 --- a/comfy/lora.py +++ b/comfy/lora.py @@ -388,6 +388,14 @@ def model_lora_keys_unet(model, key_map={}): key_map["{}".format(key_lora)] = k if isinstance(model, comfy.model_base.HunyuanVideo): + diffusers_keys = comfy.utils.flux_to_diffusers(model.model_config.unet_config, output_prefix="diffusion_model.") + for j in diffusers_keys: + if j.endswith(".weight"): + to = diffusers_keys[j] + key_map["transformer.{}".format(j[:-len(".weight")])] = to #simpletrainer and probably regular diffusers flux lora format + key_map["lycoris_{}".format(j[:-len(".weight")].replace(".", "_"))] = to #simpletrainer lycoris + key_map["lora_transformer_{}".format(j[:-len(".weight")].replace(".", "_"))] = to #onetrainer + for k in sdk: if k.startswith("diffusion_model.") and k.endswith(".weight"): # diffusion-pipe lora format @@ -398,7 +406,7 @@ def model_lora_keys_unet(model, key_map={}): key_lora = key_lora[len("diffusion_model."):-len(".weight")] key_map["transformer.{}".format(key_lora)] = k key_map["diffusion_model.{}".format(key_lora)] = k # Old loras - + if isinstance(model, comfy.model_base.LTXV): for k in sdk: if k.startswith("diffusion_model.") and k.endswith(".weight"):