pin torch

IlyasMoutawwakil · IlyasMoutawwakil · commit 03d678967106 · 2024-04-25T09:44:10.000+02:00
diff --git a/setup.py b/setup.py
@@ -27,7 +27,7 @@
     assert False, "Error: Could not open '%s' due %s\n" % (filepath, error)
 
 INSTALL_REQUIRE = [
-    "torch>=1.11",
+    "torch>=1.11,<2.3",
     "transformers>=4.36.0,<4.41.0",
     "optimum~=1.19",
     "datasets>=1.4.0",
