diff --git a/setup.py b/setup.py index 57acaddf5a..bd2235ec65 100644 --- a/setup.py +++ b/setup.py @@ -54,7 +54,7 @@ install_requires = [ 'mosaicml[libcloud,wandb,oci,gcs,mlflow]>=0.28.0,<0.29', 'mlflow>=2.14.1,<2.19', - 'accelerate>=0.25,<1.2', # for HF inference `device_map` + 'accelerate>=0.25,<1.3', # for HF inference `device_map` 'transformers>=4.43.2,<4.47', 'mosaicml-streaming>=0.10.0,<0.11', 'torch>=2.5.1,<2.5.2',