torch < 2.4.1

databricks · Aug 21, 2024 · a1f05a0 · a1f05a0
1 parent 818c66a
commit a1f05a0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -3,7 +3,7 @@
 
 # build requirements
 [build-system]
-requires = ["setuptools < 70.0.0", "torch >= 2.3.0, < 2.5"]
+requires = ["setuptools < 70.0.0", "torch >= 2.3.0, < 2.4.1"]
 build-backend = "setuptools.build_meta"
 
 # Pytest

diff --git a/setup.py b/setup.py
@@ -62,7 +62,7 @@
 install_requires = [
     'numpy>=1.21.5,<2.1.0',
     'packaging>=21.3.0,<24.2',
-    'torch>=2.3.0,<2.5',
+    'torch>=2.3.0,<2.4.1',
     'triton>=2.1.0',
     'stanford-stk @ git+https://[email protected]/eitanturok/stk.git@bump-version',
 ]