From b18ab5fd7a6b949e7ac0739ddd4777e9383a5319 Mon Sep 17 00:00:00 2001 From: Kai Fricke Date: Tue, 13 Jun 2023 10:19:43 +0100 Subject: [PATCH] [train] Bump accelerate version Signed-off-by: Kai Fricke --- python/requirements/ml/requirements_train.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/requirements/ml/requirements_train.txt b/python/requirements/ml/requirements_train.txt index 5f999b3648f4..d07f8f21df9e 100644 --- a/python/requirements/ml/requirements_train.txt +++ b/python/requirements/ml/requirements_train.txt @@ -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