Limit transformers version for tf models ONNX export (#1766)

huggingface · Mar 22, 2024 · 1146eae · 1146eae
1 parent a06d8b7
commit 1146eae
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -72,6 +72,7 @@
         "timm",
         "h5py",
         "numpy<1.24.0",
+        "transformers[sentencepiece]>=4.26.0,<4.38.0",
     ],
     "diffusers": ["diffusers"],
     "intel": "optimum-intel>=1.15.0",