File size: 209 Bytes
bc8c4af | 1 2 3 4 5 6 7 | def AnimaDiTStateDictConverter(state_dict):
new_state_dict = {}
for key in state_dict:
value = state_dict[key]
new_state_dict[key.replace("net.", "")] = value
return new_state_dict
|