Pin PyTorch to make CI green

2021-10-21 11:59:23 -04:00
parent f9c16b02e3
commit 0f502682fb
2 changed files with 2 additions and 2 deletions
--- a/setup.py
+++ b/setup.py
@@ -149,7 +149,7 @@ _deps = [
    "timeout-decorator",
    "timm",
    "tokenizers>=0.10.1,<0.11",
-    "torch>=1.0",
+    "torch>=1.0,<1.10",
    "torchaudio",
    "tqdm>=4.27",
    "unidic>=1.0.2",