dnrops
/
transformers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Playing around with versions to see if this helps

This commit is contained in:
Matt 2024-01-30 18:49:43 +00:00
parent 05fa5e0bab
commit 039ec4b5ef
3 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/pytorch/_tests_requirements.txt
View File

@ -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

2
setup.py
View File

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

2
src/transformers/dependency_versions_table.py
View File

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