Update setup.py (#76)

google-ai-edge · Jul 2, 2024 · 55fb301 · 55fb301
1 parent 3f304d4
commit 55fb301
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -74,6 +74,6 @@
         "scipy",
         "safetensors",
         "tabulate",
-        "torch==2.4.*",  # 2.4.0 stable release does not exist. Force using torch nightly.
+        "torch>=2.4.0",  # 2.4.0 stable release does not exist. Force using torch nightly.
     ],
 )