mirror of
https://github.com/modelscope/DiffSynth-Studio.git
synced 2026-03-18 22:08:13 +00:00
8 lines
232 B
Python
8 lines
232 B
Python
import torch
|
|
|
|
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 |