torch<3

stanford-futuredata · Aug 21, 2024 · 5d6423e · 5d6423e
1 parent 8e8211b
commit 5d6423e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -1,7 +1,7 @@
 from setuptools import find_packages, setup
 
 install_requires = [
-    'torch>=2.3.0,<2.4.1',
+    'torch>=2.3.0,<3.0',
     'triton>=2.1.0',
 ]