From 0f942181a42d2d7207a4f7f5d45f93e36738dd19 Mon Sep 17 00:00:00 2001 From: Casper Date: Sun, 24 Mar 2024 12:24:51 +0100 Subject: [PATCH] Pin transformers>=4.35.0,<=4.38.2 (#408) --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 409f73d7..a46b9443 100644 --- a/setup.py +++ b/setup.py @@ -86,7 +86,7 @@ def get_kernels_whl_url( requirements = [ "torch>=2.0.1", - "transformers>=4.35.0", + "transformers>=4.35.0,<=4.38.2", "tokenizers>=0.12.1", "typing_extensions>=4.8.0", "accelerate",