Pin Torch to <2.2.0 (#28785)

* Pin torch to <2.2.0 * Pin torchvision and torchaudio as well * Playing around with versions to see if this helps * twiddle something to restart the CI * twiddle it back * Try changing the natten version * make fixup * Revert "Try changing the natten version" This reverts commit de0d6592c35dc39ae8b5a616c27285db28262d06. * make fixup * fix fix fix * fix fix fix --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2024-01-30 22:01:12 +00:00
parent 415e9a0980
commit 74c9cfeaa7
4 changed files with 10 additions and 7 deletions
--- a/setup.py
+++ b/setup.py
@@ -175,9 +175,9 @@ _deps = [
    "timeout-decorator",
    "timm",
    "tokenizers>=0.14,<0.19",
-    "torch>=1.11,!=1.12.0",
-    "torchaudio",
-    "torchvision",
+    "torch<2.2.0",
+    "torchaudio<2.2.0",
+    "torchvision<0.17.0",
    "pyctcdecode>=0.4.0",
    "tqdm>=4.27",
    "unidic>=1.0.2",