Pin PyTorch while we fix compatibility with 1.12

2022-06-28 15:07:26 -04:00
parent 6c8f4c9a93
commit 5a3d0cbdda
2 changed files with 2 additions and 2 deletions
--- a/setup.py
+++ b/setup.py
@@ -160,7 +160,7 @@ _deps = [
    "timeout-decorator",
    "timm",
    "tokenizers>=0.11.1,!=0.11.3,<0.13",
-    "torch>=1.0",
+    "torch>=1.0,<1.12",
    "torchaudio",
    "pyctcdecode>=0.3.0",
    "tqdm>=4.27",