Commit 0e892ad6 authored by Woosuk Kwon's avatar Woosuk Kwon
Browse files

Upgrade to PyTorch 2.2.1

parent a43fbbf1
...@@ -294,6 +294,6 @@ setup( ...@@ -294,6 +294,6 @@ setup(
"bdist_wheel": CachedWheelsCommand, "bdist_wheel": CachedWheelsCommand,
}, },
python_requires=">=3.8", python_requires=">=3.8",
install_requires=["torch == 2.1.2"], install_requires=["torch == 2.2.1"],
setup_requires=["psutil"], setup_requires=["psutil"],
) )
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment