mirror of
https://github.com/modelscope/DiffSynth-Studio.git
synced 2026-03-23 09:28:12 +00:00
DiffSynth-Studio 2.0 major update
This commit is contained in:
15
diffsynth/utils/state_dict_converters/nexus_gen_projector.py
Normal file
15
diffsynth/utils/state_dict_converters/nexus_gen_projector.py
Normal file
@@ -0,0 +1,15 @@
|
||||
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
|
||||
Reference in New Issue
Block a user