diff --git a/requirements-test.txt b/requirements-test.txt index 4905710..6dc578d 100644 --- a/requirements-test.txt +++ b/requirements-test.txt @@ -3,7 +3,7 @@ transformers==4.27.2 fastapi==0.95.0 uvicorn==0.21.1 nltk==3.8.1 -torch==2.0.0 +torch==2.0.1 sentencepiece==0.1.97 sentence-transformers==2.2.2 pytest diff --git a/requirements.txt b/requirements.txt index 7b0986b..5091cd0 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,6 +2,6 @@ transformers==4.34.0 fastapi==0.103.2 uvicorn==0.23.2 nltk==3.8.1 -torch==2.1.0 +torch==2.0.1 sentencepiece==0.1.99 sentence-transformers==2.2.2