Skip to content

Commit a1fce9e

Browse files
Update accelerate requirement from <1.4,>=0.25 to >=0.25,<1.7 (#1799)
Signed-off-by: dependabot[bot] <[email protected]> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
1 parent ba6e43c commit a1fce9e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@
5454
install_requires = [
5555
'mosaicml[libcloud,wandb,oci,gcs,mlflow]>=0.30.0,<0.31',
5656
'mlflow>=2.14.1,<2.22',
57-
'accelerate>=0.25,<1.4', # for HF inference `device_map`
57+
'accelerate>=0.25,<1.7', # for HF inference `device_map`
5858
'transformers>=v4.51.0,<4.52',
5959
'mosaicml-streaming>=0.12.0,<0.13',
6060
'torch>=2.6.0,<2.6.1',

0 commit comments

Comments
 (0)