File size: 257 Bytes
bc8c4af | 1 2 3 4 5 6 7 | 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_ |