"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "b6404866cda2952942a39e424206aed0cd6aeb5c"
Unverified Commit 45e3d649 authored by Zachary Mueller's avatar Zachary Mueller Committed by GitHub
Browse files

Update error message when Accelerate isn't installed (#23373)

Update error
parent ea0eb156
...@@ -1611,7 +1611,7 @@ class TrainingArguments: ...@@ -1611,7 +1611,7 @@ class TrainingArguments:
logger.info("PyTorch: setting up devices") logger.info("PyTorch: setting up devices")
if not is_sagemaker_mp_enabled() and not is_accelerate_available(check_partial_state=True): if not is_sagemaker_mp_enabled() and not is_accelerate_available(check_partial_state=True):
raise ImportError( raise ImportError(
"Using the `Trainer` with `PyTorch` requires `accelerate`: Run `pip install --upgrade accelerate`" "Using the `Trainer` with `PyTorch` requires `accelerate>=0.19.0`: Please run `pip install transformers[torch]` or `pip install accelerate -U`"
) )
if self.no_cuda: if self.no_cuda:
self.distributed_state = PartialState(cpu=True, backend=self.ddp_backend) self.distributed_state = PartialState(cpu=True, backend=self.ddp_backend)
......
...@@ -502,7 +502,7 @@ def is_protobuf_available(): ...@@ -502,7 +502,7 @@ def is_protobuf_available():
def is_accelerate_available(check_partial_state=False): def is_accelerate_available(check_partial_state=False):
if check_partial_state: if check_partial_state:
return _accelerate_available and version.parse(_accelerate_version) >= version.parse("0.17.0") return _accelerate_available and version.parse(_accelerate_version) >= version.parse("0.19.0")
return _accelerate_available return _accelerate_available
......
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