From d49e6a27f4591c0572f3509c508092ed4af6b571 Mon Sep 17 00:00:00 2001 From: Daniel King <43149077+dakinggg@users.noreply.github.com> Date: Tue, 23 Jul 2024 11:36:57 -0700 Subject: [PATCH] bump transformers (#1388) --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index b4948e9a43..4563b4dfb8 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.42.3,<4.43', + 'transformers>=4.43.1,<4.44', 'mosaicml-streaming>=0.7.6,<0.8', 'torch>=2.3.0,<2.4', 'datasets>=2.19,<2.20',