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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

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: