diff --git a/setup.py b/setup.py index a62fc46e..64baf488 100644 --- a/setup.py +++ b/setup.py @@ -47,7 +47,7 @@ setuptools.setup( url="https://github.com/stanford-oval/genienlp", install_requires=[ 'numpy>=1.14.5', - 'torch>=1.9.0,<1.13.0', + 'torch>=1.9.0,<1.14.0', 'transformers==4.22.2', 'datasets==2.6.1', 'pathos==0.2.8',