diff --git a/setup.py b/setup.py index 7b8e295..b46182d 100644 --- a/setup.py +++ b/setup.py @@ -3,7 +3,7 @@ from setuptools import find_packages, setup requires = [ - "torch==1.8.0", + "torch==2.2.0", "pytorch-nlp==0.5.0", "pytorch-pfn-extras==0.3.2", "requests",