mirror of
https://github.com/modelscope/DiffSynth-Studio.git
synced 2026-03-19 14:58:12 +00:00
5 lines
210 B
Python
5 lines
210 B
Python
def FluxTextEncoderT5StateDictConverter(state_dict):
|
|
state_dict_ = {i: state_dict[i] for i in state_dict}
|
|
state_dict_["encoder.embed_tokens.weight"] = state_dict["shared.weight"]
|
|
return state_dict_
|