bump torch to <2.5

databricks · Aug 9, 2024 · e0ff227 · e0ff227
1 parent 6adb1fb
commit e0ff227
Showing 1 changed file with 1 addition and 1 deletion.
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.4',
+    'torch>=2.3.0,<2.5',
     'triton>=2.1.0',
     'stanford-stk @ git+https://[email protected]/stanford-futuredata/stk.git@a1ddf98466730b88a2988860a9d8000fd1833301',
 ]