diff --git a/requirements.txt b/requirements.txt index 6cb79d3..9d4e4e1 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ numpy>=1.13.3 torch>=0.4.0 -transformers>=3.5.1,<4.0.0 +transformers>=3.5.1,<5.0.0 sklearn diff --git a/requirements_rtx30.txt b/requirements_rtx30.txt index d1378bb..bcec6b9 100644 --- a/requirements_rtx30.txt +++ b/requirements_rtx30.txt @@ -1,5 +1,5 @@ numpy==1.20.1 torch==1.7.1+cu110 --find-links https://download.pytorch.org/whl/torch_stable.html -transformers==3.5.1 +transformers==4.30.0 sklearn