mirror of
https://github.com/modelscope/DiffSynth-Studio.git
synced 2026-03-24 01:48:13 +00:00
Add: FLUX
This commit is contained in:
8
diffsynth/utils/state_dict_converters/nexus_gen.py
Normal file
8
diffsynth/utils/state_dict_converters/nexus_gen.py
Normal file
@@ -0,0 +1,8 @@
|
||||
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
|
||||
Reference in New Issue
Block a user