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

unpin torch (#28892)

Browse Source 
* unpin torch

* check

* check

* check

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2024-02-06 17:21:05 +01:00
committed by GitHub
parent 76b4f666f5
commit 89439fea64
3 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.circleci/create_circleci_config.py
View File

@@ -283,6 +283,8 @@ torch_and_tf_job = CircleCIJob(
"pip install -U --upgrade-strategy eager .[sklearn,tf-cpu,torch,testing,sentencepiece,torch-speech,vision]", "pip install -U --upgrade-strategy eager .[sklearn,tf-cpu,torch,testing,sentencepiece,torch-speech,vision]",
"pip install -U --upgrade-strategy eager tensorflow_probability", "pip install -U --upgrade-strategy eager tensorflow_probability",
"pip install -U --upgrade-strategy eager -e git+https://github.com/huggingface/accelerate@main#egg=accelerate", "pip install -U --upgrade-strategy eager -e git+https://github.com/huggingface/accelerate@main#egg=accelerate",
# TODO: remove this one after fixing the dependency issue(s) above
"pip install -U --upgrade-strategy eager torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu"
], ],
marker="is_pt_tf_cross_test", marker="is_pt_tf_cross_test",
pytest_options={"rA": None, "durations": 0}, pytest_options={"rA": None, "durations": 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<2.2.0", "torch",
"torchaudio<2.2.0", "torchaudio",
"torchvision<0.17.0", "torchvision",
"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<2.2.0", "torch": "torch",
"torchaudio": "torchaudio<2.2.0", "torchaudio": "torchaudio",
"torchvision": "torchvision<0.17.0", "torchvision": "torchvision",
"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",