diff --git a/setup.py b/setup.py index 4563b4dfb8..6193fdc6b4 100644 --- a/setup.py +++ b/setup.py @@ -57,7 +57,7 @@ '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` - 'transformers>=4.43.1,<4.44', + 'transformers>=4.43.2,<4.44', 'mosaicml-streaming>=0.7.6,<0.8', 'torch>=2.3.0,<2.4', 'datasets>=2.19,<2.20',