Merge pull request #434 from sanchit97/fix-torch-version

Fix torch version to 1.7.1
QData · Mar 21, 2021 · e343f6d · e343f6d
2 parents 54cbfbc + 93a8c5c
commit e343f6d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -10,7 +10,7 @@ nltk
 numpy<1.19.0 #TF 2.0 requires this
 pandas>=1.0.1
 scipy==1.4.1
-torch
+torch==1.7.1
 transformers>=3.3.0
 terminaltables
 tqdm>=4.27,<4.50.0