Support ModelScope-Trainer DiffSynth lora for Z Image. (#11805)

This commit is contained in:
kelseyee
2026-01-13 04:38:46 +08:00
committed by GitHub
parent c6238047ee
commit a3b5d4996a

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: