mirror of
https://github.com/modelscope/DiffSynth-Studio.git
synced 2026-03-19 06:23:43 +00:00
13 lines
588 B
Python
13 lines
588 B
Python
def WanS2VAudioEncoderStateDictConverter(state_dict):
|
|
rename_dict = {
|
|
"model.wav2vec2.encoder.pos_conv_embed.conv.weight_g": "model.wav2vec2.encoder.pos_conv_embed.conv.parametrizations.weight.original0",
|
|
"model.wav2vec2.encoder.pos_conv_embed.conv.weight_v": "model.wav2vec2.encoder.pos_conv_embed.conv.parametrizations.weight.original1",
|
|
}
|
|
state_dict_ = {}
|
|
for name in state_dict:
|
|
name_ = "model." + name
|
|
if name_ in rename_dict:
|
|
name_ = rename_dict[name_]
|
|
state_dict_[name_] = state_dict[name]
|
|
return state_dict_
|