diff --git a/setup.py b/setup.py old mode 100755 new mode 100644 index 439590b..4bf31a7 --- a/setup.py +++ b/setup.py @@ -8,7 +8,7 @@ packages=find_packages(), python_requires=">=3.6", install_requires=[ - "torch==1.4.0", + "torch==2.2.0", "numpy==1.18.1", "jax==0.1.68", "jaxlib==0.1.47",