diff --git a/requirements.txt b/requirements.txt index c45c129..9bb4b3e 100644 --- a/requirements.txt +++ b/requirements.txt @@ -11,7 +11,7 @@ truecase requests>=2.20 pandas==1.3.5 transformers==4.33.2 -torch==1.10.1 +torch==1.13.1 pytest tensorflow-addons==0.19.0 blingfire==0.1.8 diff --git a/setup.py b/setup.py index b44d0ce..2b78335 100644 --- a/setup.py +++ b/setup.py @@ -23,7 +23,7 @@ 'pydot==1.4.0', 'lmdb==1.2.1', 'transformers==4.33.2', - 'torch==1.10.1', + 'torch==1.13.1', 'truecase', 'requests>=2.20', 'pandas==1.3.5',