Upgrade to PyTorch 2.2.1

0e892ad6 · Woosuk Kwon · a43fbbf1 · 0e892ad6
Commit 0e892ad6 authored Apr 22, 2024 by Woosuk Kwon
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

setup.py setup.py +1 -1

No files found.
--- a/setup.py
+++ b/setup.py
@@ -294,6 +294,6 @@ setup(
        "bdist_wheel": CachedWheelsCommand,
    },
    python_requires=">=3.8",
-    install_requires=["torch == 2.1.2"],
+    install_requires=["torch == 2.2.1"],
    setup_requires=["psutil"],
 )