third_party/DiffSynth-Studio/diffsynth/utils/state_dict_converters/anima_dit.py

def AnimaDiTStateDictConverter(state_dict):
    new_state_dict = {}
    for key in state_dict:
        value = state_dict[key]
        new_state_dict[key.replace("net.", "")] = value
    return new_state_dict