Unverified Commit 396a6a2e authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Fix minimum version for device_map (#20489)

parent 08b46218
...@@ -1962,8 +1962,8 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix ...@@ -1962,8 +1962,8 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix
# low_cpu_mem_usage requires PyTorch >= 1.9 to have the meta device. # low_cpu_mem_usage requires PyTorch >= 1.9 to have the meta device.
require_version_core("torch>=1.9") require_version_core("torch>=1.9")
if device_map is not None: if device_map is not None:
# The max memory utils require PyTorch >= 1.11 to have torch.cuda.mem_get_info. # The max memory utils require PyTorch >= 1.10 to have torch.cuda.mem_get_info.
require_version_core("torch>=1.11") require_version_core("torch>=1.10")
if is_deepspeed_zero3_enabled(): if is_deepspeed_zero3_enabled():
raise ValueError( raise ValueError(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment