PyTorch >= 1.7.0 and TensorFlow >= 2.4.0 (#19016)

2022-09-14 07:19:02 -04:00
parent 9f4acd059f
commit a2a3afbc8d
30 changed files with 111 additions and 325 deletions
--- a/setup.py
+++ b/setup.py
@@ -155,13 +155,13 @@ _deps = [
    "librosa",
    "starlette",
    "tensorflow-cpu>=2.3",
-    "tensorflow>=2.3",
+    "tensorflow>=2.4",
    "tensorflow-text",
    "tf2onnx",
    "timeout-decorator",
    "timm",
    "tokenizers>=0.11.1,!=0.11.3,<0.13",
-    "torch>=1.0,!=0.12.0",
+    "torch>=1.7,!=1.12.0",
    "torchaudio",
    "pyctcdecode>=0.3.0",
    "tqdm>=4.27",