mirror of
https://github.com/modelscope/DiffSynth-Studio.git
synced 2026-03-23 17:38:10 +00:00
4 lines
172 B
Python
4 lines
172 B
Python
def ZImageDiTStateDictConverter(state_dict):
|
|
state_dict_ = {name.replace("model.diffusion_model.", ""): state_dict[name] for name in state_dict}
|
|
return state_dict_
|