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_