Function update_state_dict_inplace

tools/convert_weight_sat2hf.py:156–157

(state_dict: Dict[str, Any], old_key: str, new_key: str)

# Function to update the state_dict with new key assignments in-place
def update_state_dict_inplace(state_dict: Dict[str, Any], old_key: str, new_key: str) -> Dict[str, Any]:
    state_dict[new_key] = state_dict.pop(old_key)


# Function to convert a transformer checkpoint to the CogVideoX format