unpin torch (#28892)

* unpin torch * check * check * check --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2024-02-06 17:21:05 +01:00
parent 76b4f666f5
commit 89439fea64
3 changed files with 8 additions and 6 deletions
--- a/setup.py
+++ b/setup.py
@@ -175,9 +175,9 @@ _deps = [
    "timeout-decorator",
    "timm",
    "tokenizers>=0.14,<0.19",
-    "torch<2.2.0",
-    "torchaudio<2.2.0",
-    "torchvision<0.17.0",
+    "torch",
+    "torchaudio",
+    "torchvision",
    "pyctcdecode>=0.4.0",
    "tqdm>=4.27",
    "unidic>=1.0.2",