mirror of
https://github.com/modelscope/DiffSynth-Studio.git
synced 2026-03-19 06:48:12 +00:00
15 lines
552 B
Python
15 lines
552 B
Python
def NexusGenMergerStateDictConverter(state_dict):
|
|
merger_state_dict = {}
|
|
for key in state_dict:
|
|
if key.startswith('embedding_merger.'):
|
|
value = state_dict[key]
|
|
new_key = key.replace("embedding_merger.", "")
|
|
merger_state_dict[new_key] = value
|
|
return merger_state_dict
|
|
|
|
def NexusGenAdapterStateDictConverter(state_dict):
|
|
adapter_state_dict = {}
|
|
for key in state_dict:
|
|
if key.startswith('adapter.'):
|
|
adapter_state_dict[key] = state_dict[key]
|
|
return adapter_state_dict |