From 7cb773d970cd9ce7ce566a193379801f74517334 Mon Sep 17 00:00:00 2001 From: "dependabot[bot]" <49699333+dependabot[bot]@users.noreply.github.com> Date: Sat, 20 Jul 2024 23:03:54 +0000 Subject: [PATCH] Update accelerate requirement from <0.26,>=0.25 to >=0.32.1,<0.33 Updates the requirements on [accelerate](https://github.com/huggingface/accelerate) to permit the latest version. - [Release notes](https://github.com/huggingface/accelerate/releases) - [Commits](https://github.com/huggingface/accelerate/compare/v0.25.0...v0.32.1) --- updated-dependencies: - dependency-name: accelerate dependency-type: direct:production ... Signed-off-by: dependabot[bot] --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index eb22d8f6a3..b4948e9a43 100644 --- a/setup.py +++ b/setup.py @@ -56,7 +56,7 @@ install_requires = [ 'mosaicml[libcloud,wandb,oci,gcs,mlflow]>=0.23.4,<0.24', 'mlflow>=2.14.1,<2.15', - 'accelerate>=0.25,<0.26', # for HF inference `device_map` + 'accelerate>=0.25,<0.33', # for HF inference `device_map` 'transformers>=4.42.3,<4.43', 'mosaicml-streaming>=0.7.6,<0.8', 'torch>=2.3.0,<2.4',