SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Pin Torch to <2.2.0 (#28785)

Browse Source 
* 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>
This commit is contained in:
Matt
2024-01-30 22:01:12 +00:00
committed by GitHub
parent 415e9a0980
commit 74c9cfeaa7
4 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
.circleci/create_circleci_config.py
View File

@@ -474,6 +474,7 @@ exotic_models_job = CircleCIJob(
"pip install -U --upgrade-strategy eager python-Levenshtein", "pip install -U --upgrade-strategy eager python-Levenshtein",
"pip install -U --upgrade-strategy eager opencv-python", "pip install -U --upgrade-strategy eager opencv-python",
"pip install -U --upgrade-strategy eager nltk", "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_to_run=[
"tests/models/*layoutlmv*", "tests/models/*layoutlmv*",

4
examples/pytorch/_tests_requirements.txt
View File

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

6
setup.py
View File

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

6
src/transformers/dependency_versions_table.py
View File

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