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",