Pin transformers>=4.35.0,<=4.38.2 (#408)

0f942181 · Casper · GitHub · 94e73f0b · 0f942181
Unverified Commit 0f942181 authored Mar 24, 2024 by Casper Committed by GitHub Mar 24, 2024
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

setup.py setup.py +1 -1

No files found.
--- a/setup.py
+++ b/setup.py
@@ -86,7 +86,7 @@ common_setup_kwargs = {

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