diff --git a/requirements.txt b/requirements.txt index 7ccd20e..26c54a5 100644 --- a/requirements.txt +++ b/requirements.txt @@ -11,6 +11,6 @@ truecase requests>=2.20 pandas==1.3.5 transformers==4.25.1 -torch==1.10.1 +torch==1.13.1 pytest tensorflow-addons==0.19.0 diff --git a/setup.py b/setup.py index 456da4c..2ca5d08 100644 --- a/setup.py +++ b/setup.py @@ -23,7 +23,7 @@ 'pydot==1.4.0', 'lmdb==1.2.1', 'transformers==4.25.1', - 'torch==1.10.1', + 'torch==1.13.1', 'truecase', 'requests>=2.20', 'pandas==1.3.5',