src/transformers/models/dpr/convert_dpr_original_checkpoint_to_pytorch.py/load_states_from_checkpoint
def load_states_from_checkpoint(model_file: str) -> CheckpointState:
    print(f"Reading saved model from {model_file}")
    state_dict = torch.load(model_file, map_location=lambda s, l: default_restore_location(s, "cpu"))
    return CheckpointState(**state_dict)