File size: 218 Bytes
bc8c4af | 1 2 3 4 5 6 | def NexusGenAutoregressiveModelStateDictConverter(state_dict):
new_state_dict = {}
for key in state_dict:
value = state_dict[key]
new_state_dict["model." + key] = value
return new_state_dict |