Function convert

scripts/convert_animatediff_sparsectrl_to_diffusers.py:21–30

(original_state_dict: Dict[str, nn.Module])

def convert(original_state_dict: Dict[str, nn.Module]) -> dict[str, nn.Module]:
    converted_state_dict = {}

    for key in list(original_state_dict.keys()):
        renamed_key = key
        for new_name, old_name in KEYS_RENAME_MAPPING.items():
            renamed_key = renamed_key.replace(new_name, old_name)
        converted_state_dict[renamed_key] = original_state_dict.pop(key)

    return converted_state_dict


def get_args():