Unverified Commit e37ab52d authored by Michael Feil's avatar Michael Feil Committed by GitHub
Browse files

Bug: `training_args.py` fix missing import with accelerate with version...

Bug: `training_args.py` fix missing import with accelerate with version `accelerate==0.20.1` (#28171)

* fix-accelerate-version

* updated with exported ACCELERATE_MIN_VERSION,

* update string in ACCELERATE_MIN_VERSION
parent c9fb250a
...@@ -36,6 +36,7 @@ from .trainer_utils import ( ...@@ -36,6 +36,7 @@ from .trainer_utils import (
SchedulerType, SchedulerType,
) )
from .utils import ( from .utils import (
ACCELERATE_MIN_VERSION,
ExplicitEnum, ExplicitEnum,
cached_property, cached_property,
is_accelerate_available, is_accelerate_available,
...@@ -1837,9 +1838,10 @@ class TrainingArguments: ...@@ -1837,9 +1838,10 @@ class TrainingArguments:
requires_backends(self, ["torch"]) requires_backends(self, ["torch"])
logger.info("PyTorch: setting up devices") logger.info("PyTorch: setting up devices")
if not is_sagemaker_mp_enabled(): if not is_sagemaker_mp_enabled():
if not is_accelerate_available(min_version="0.20.1"): if not is_accelerate_available():
raise ImportError( raise ImportError(
"Using the `Trainer` with `PyTorch` requires `accelerate>=0.20.1`: Please run `pip install transformers[torch]` or `pip install accelerate -U`" f"Using the `Trainer` with `PyTorch` requires `accelerate>={ACCELERATE_MIN_VERSION}`: "
"Please run `pip install transformers[torch]` or `pip install accelerate -U`"
) )
AcceleratorState._reset_state(reset_partial_state=True) AcceleratorState._reset_state(reset_partial_state=True)
self.distributed_state = None self.distributed_state = None
......
...@@ -90,6 +90,7 @@ from .hub import ( ...@@ -90,6 +90,7 @@ from .hub import (
try_to_load_from_cache, try_to_load_from_cache,
) )
from .import_utils import ( from .import_utils import (
ACCELERATE_MIN_VERSION,
ENV_VARS_TRUE_AND_AUTO_VALUES, ENV_VARS_TRUE_AND_AUTO_VALUES,
ENV_VARS_TRUE_VALUES, ENV_VARS_TRUE_VALUES,
TORCH_FX_REQUIRED_VERSION, TORCH_FX_REQUIRED_VERSION,
......
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