compatibility patch

This commit is contained in:
Artiprocher
2026-03-23 11:24:19 +08:00
parent 078fc551d9
commit 0d466231d1
2 changed files with 10 additions and 0 deletions

View File

@@ -0,0 +1,3 @@
def ZImageDiTStateDictConverter(state_dict):
state_dict_ = {name.replace("model.diffusion_model.", ""): state_dict[name] for name in state_dict}
return state_dict_