diff --git a/comfy/lora.py b/comfy/lora.py index 63ee85323..f6f1fb150 100644 --- a/comfy/lora.py +++ b/comfy/lora.py @@ -328,6 +328,13 @@ def model_lora_keys_unet(model, key_map={}): key_map["lycoris_{}".format(key_lora.replace(".", "_"))] = to key_map[key_lora] = to + # TwinFlow-Z-Image LoRAs can target t_embedder_2.* keys. + # Alias them back to t_embedder.* targets for compatibility. + if isinstance(model, comfy.model_base.TwinFlow_Z_Image): + for key in list(key_map.keys()): + if "t_embedder." in key and "t_embedder_2." not in key: + key_map[key.replace("t_embedder.", "t_embedder_2.", 1)] = key_map[key] + if isinstance(model, comfy.model_base.Kandinsky5): for k in sdk: if k.startswith("diffusion_model.") and k.endswith(".weight"):