diff --git a/setup.py b/setup.py
index 494db8d62a..2ad11ad994 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.33',  # for HF inference `device_map`
+    'accelerate>=0.25,<0.34',  # for HF inference `device_map`
     'transformers>=4.43.2,<4.44',
     'mosaicml-streaming>=0.8.0,<0.9',
     'torch>=2.3.0,<2.4',