DiffSynth-Studio/diffsynth/utils/state_dict_converters/wans2v_audio_encoder.py

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_