Merge pull request #15 from eitanturok/bump-version

bump torch, stk versions
stanford-futuredata · Aug 21, 2024 · 7363137 · 7363137
2 parents 7b8f0df + 5d6423e
commit 7363137
Showing 1 changed file with 2 additions and 2 deletions.
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',
+    'torch>=2.3.0,<3.0',
     'triton>=2.1.0',
 ]
 
@@ -15,7 +15,7 @@
 
 setup(
     name="stanford-stk",
-    version="0.7.0",
+    version="0.7.1",
     author="Trevor Gale",
     author_email="[email protected]",
     description="Sparse Toolkit",