Support ModelScope-Trainer DiffSynth lora for Z Image.

This commit is contained in:
kelseyee 2026-01-11 19:34:36 +08:00
parent 5cd1113236
commit 78e5c2eaca

View File

@ -322,6 +322,7 @@ def model_lora_keys_unet(model, key_map={}):
key_map["diffusion_model.{}".format(key_lora)] = to
key_map["transformer.{}".format(key_lora)] = to
key_map["lycoris_{}".format(key_lora.replace(".", "_"))] = to
key_map[key_lora] = to
if isinstance(model, comfy.model_base.Kandinsky5):
for k in sdk: