mirror of
https://github.com/modelscope/DiffSynth-Studio.git
synced 2026-03-19 14:58:12 +00:00
7 lines
257 B
Python
7 lines
257 B
Python
def WanVideoVAEStateDictConverter(state_dict):
|
|
state_dict_ = {}
|
|
if 'model_state' in state_dict:
|
|
state_dict = state_dict['model_state']
|
|
for name in state_dict:
|
|
state_dict_['model.' + name] = state_dict[name]
|
|
return state_dict_ |