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>

.circleci/create_circleci_config.py

@@ -474,6 +474,7 @@ exotic_models_job = CircleCIJob(
         "pip install -U --upgrade-strategy eager python-Levenshtein",
         "pip install -U --upgrade-strategy eager opencv-python",
         "pip install -U --upgrade-strategy eager nltk",
+        "pip uninstall -y torch torchvision torchaudio && pip install -U --upgrade-strategy eager 'torch<2.2.0' 'torchvision<0.17' 'torchaudio<2.2.0'"
     ],
     tests_to_run=[
         "tests/models/*layoutlmv*",

examples/pytorch/_tests_requirements.txt

@@ -19,7 +19,9 @@ pytest
 conllu
 sentencepiece != 0.1.92
 protobuf
-torchvision
+torch<2.2.0
+torchvision<0.17
+torchaudio<2.2.0
 jiwer
 librosa
 evaluate >= 0.2.0

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",

src/transformers/dependency_versions_table.py

@@ -80,9 +80,9 @@ deps = {
     "timeout-decorator": "timeout-decorator",
     "timm": "timm",
     "tokenizers": "tokenizers>=0.14,<0.19",
-    "torch": "torch>=1.11,!=1.12.0",
-    "torchaudio": "torchaudio",
-    "torchvision": "torchvision",
+    "torch": "torch<2.2.0",
+    "torchaudio": "torchaudio<2.2.0",
+    "torchvision": "torchvision<0.17.0",
     "pyctcdecode": "pyctcdecode>=0.4.0",
     "tqdm": "tqdm>=4.27",
     "unidic": "unidic>=1.0.2",