[Versioning] Fix torch deps (pytorch#2340)

ai4co · Aug 2, 2024 · 812f936 · 812f936
1 parent c2c487b
commit 812f936
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -72,10 +72,10 @@ def _get_pytorch_version(is_nightly, is_local):
     # if "PYTORCH_VERSION" in os.environ:
     #     return f"torch=={os.environ['PYTORCH_VERSION']}"
     if is_nightly:
-        return "torch>=2.4.0.dev"
+        return "torch>=2.5.0.dev"
     elif is_local:
         return "torch"
-    return "torch>=2.3.0"
+    return "torch>=2.4.0"
 
 
 def _get_packages():