Set torch upper bound to <2.1.0

Some changes in PyTorch 2.1.0 and later are incompatible with Curated Transformers 1.x. Fixing these issues would require API changes. So we set the upper bound on supported PyTorch versions. We will soon release Curated Transformers 2.0.0, which is compatible with the lastest PyTorch versions.
explosion · Feb 9, 2024 · dadf5bb · dadf5bb
1 parent cd53833
commit dadf5bb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,7 @@
 curated-tokenizers>=0.9.1,<1.0.0
 huggingface-hub>=0.14
 tokenizers>=0.13.3
-torch>=1.12.0
+torch>=1.12.0,<2.1.0
 
 # Development dependencies
 mypy>=0.990,<1.1.0; platform_machine != "aarch64"

diff --git a/setup.cfg b/setup.cfg
@@ -17,7 +17,7 @@ install_requires =
     curated-tokenizers>=0.9.1,<1.0.0
     huggingface-hub>=0.14
     tokenizers>=0.13.3
-    torch>=1.12.0
+    torch>=1.12.0,<2.1.0
 
 [options.extras_require]
 quantization =