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

Update error
This commit is contained in:
Zachary Mueller
2023-05-17 11:16:02 -04:00
committed by GitHub
parent ea0eb15649
commit 45e3d6496a
2 changed files with 2 additions and 2 deletions

View File

@@ -1611,7 +1611,7 @@ class TrainingArguments:
logger.info("PyTorch: setting up devices")
if not is_sagemaker_mp_enabled() and not is_accelerate_available(check_partial_state=True):
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:
self.distributed_state = PartialState(cpu=True, backend=self.ddp_backend)

View File

@@ -502,7 +502,7 @@ def is_protobuf_available():
def is_accelerate_available(check_partial_state=False):
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