diff --git a/setup.py b/setup.py index 311d3da736..576fb25883 100644 --- a/setup.py +++ b/setup.py @@ -55,7 +55,7 @@ '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` - 'transformers>=4.43.2,<4.47', + 'transformers>=4.43.2,<4.48', 'mosaicml-streaming>=0.10.0,<0.11', 'torch>=2.5.1,<2.5.2', 'datasets>=2.20.0,<3.2',