From 45e3d6496a9e64c5359eee979b3e4e8b96a8de16 Mon Sep 17 00:00:00 2001 From: Zachary Mueller Date: Wed, 17 May 2023 11:16:02 -0400 Subject: [PATCH] Update error message when Accelerate isn't installed (#23373) Update error --- src/transformers/training_args.py | 2 +- src/transformers/utils/import_utils.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/transformers/training_args.py b/src/transformers/training_args.py index 44f28ff99e..3189c5f86b 100644 --- a/src/transformers/training_args.py +++ b/src/transformers/training_args.py @@ -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) diff --git a/src/transformers/utils/import_utils.py b/src/transformers/utils/import_utils.py index 413aa5ecfa..037a0d96a1 100644 --- a/src/transformers/utils/import_utils.py +++ b/src/transformers/utils/import_utils.py @@ -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