def _set_default_torch_dtype(dtype: torch.dtype):
    """Sets the default torch dtype to the given dtype."""
    old_dtype = torch.get_default_dtype()
    torch.set_default_dtype(dtype)
    yield
    torch.set_default_dtype(old_dtype)
Sets the default torch dtype to the given dtype.
verl/third_party/vllm/vllm_v_0_3_1/model_loader.py