Skip to content

Commit

Permalink
[train] Bump accelerate version (#36369)
Browse files Browse the repository at this point in the history
Resolves failing huggingface example on master.

Signed-off-by: Kai Fricke <kai@anyscale.com>
  • Loading branch information
krfricke authored Jun 13, 2023
1 parent a569a32 commit 3d0e46e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/requirements/ml/requirements_train.txt
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ transformers==4.19.1; python_version > '3.6'
# We need accelerate>=0.17.0 for AccelerateTrainer to work (not available on py 3.6)
# If changing the version here, also change it in AccelerateTrainer docstring
accelerate==0.5.1; python_version <= '3.6'
accelerate==0.17.1; python_version > '3.6'
accelerate==0.20.3; python_version > '3.6'
# Tracking issue: https://github.com/ray-project/ray/issues/34399
deepspeed==0.8.3; python_version > '3.6'
datasets==2.0.0
Expand Down

0 comments on commit 3d0e46e

Please sign in to comment.