mirror of
https://github.com/modelscope/DiffSynth-Studio.git
synced 2026-03-19 06:23:43 +00:00
33 lines
1.3 KiB
Python
33 lines
1.3 KiB
Python
def LTX2AudioEncoderStateDictConverter(state_dict):
|
|
# Not used
|
|
state_dict_ = {}
|
|
for name in state_dict:
|
|
if name.startswith("audio_vae.encoder."):
|
|
new_name = name.replace("audio_vae.encoder.", "")
|
|
state_dict_[new_name] = state_dict[name]
|
|
elif name.startswith("audio_vae.per_channel_statistics."):
|
|
new_name = name.replace("audio_vae.per_channel_statistics.", "per_channel_statistics.")
|
|
state_dict_[new_name] = state_dict[name]
|
|
return state_dict_
|
|
|
|
|
|
def LTX2AudioDecoderStateDictConverter(state_dict):
|
|
state_dict_ = {}
|
|
for name in state_dict:
|
|
if name.startswith("audio_vae.decoder."):
|
|
new_name = name.replace("audio_vae.decoder.", "")
|
|
state_dict_[new_name] = state_dict[name]
|
|
elif name.startswith("audio_vae.per_channel_statistics."):
|
|
new_name = name.replace("audio_vae.per_channel_statistics.", "per_channel_statistics.")
|
|
state_dict_[new_name] = state_dict[name]
|
|
return state_dict_
|
|
|
|
|
|
def LTX2VocoderStateDictConverter(state_dict):
|
|
state_dict_ = {}
|
|
for name in state_dict:
|
|
if name.startswith("vocoder."):
|
|
new_name = name[len("vocoder."):]
|
|
state_dict_[new_name] = state_dict[name]
|
|
return state_dict_
|