diff --git a/setup.py b/setup.py index 07a0bea4a8..b07b8afe08 100644 --- a/setup.py +++ b/setup.py @@ -49,7 +49,7 @@ install_requires = [ 'mosaicml[libcloud,wandb,mlflow]>=0.16.1,<0.17', 'accelerate>=0.20,<0.21', # for HF inference `device_map` - 'transformers>=4.32,<4.33', + 'transformers>=4.33,<4.34', 'mosaicml-streaming>=0.5.1,<0.6', 'torch>=1.13.1,<2.1.1', 'datasets>=2.14.5,<2.15',