compatibility patch (#1363)

This commit is contained in:
Zhongjie Duan
2026-03-23 11:24:49 +08:00
committed by GitHub
parent 078fc551d9
commit 5bccd60c80
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_