diff --git a/requirements.txt b/requirements.txt index b5255d03..b4539c53 100644 --- a/requirements.txt +++ b/requirements.txt @@ -13,7 +13,7 @@ scikit-learn scipy==1.4.1 sentence_transformers>0.2.6 stanza -torch +torch==1.6 transformers==3.3.0 tensorflow>=2 tensorflow_hub