use correct deps for torchhub (#9552)

This commit is contained in:
Stas Bekman
2021-01-13 05:02:53 -08:00
committed by GitHub
parent eabad8fd9c
commit b2dfcc567b
2 changed files with 7 additions and 3 deletions

View File

@@ -1,6 +1,6 @@
name: Torch hub integration
on:
on:
push:
branches:
- "*"
@@ -32,8 +32,10 @@ jobs:
- name: Install dependencies
run: |
pip install --upgrade pip
pip install torch
pip install numpy filelock protobuf requests tqdm regex sentencepiece sacremoses tokenizers==0.9.4 packaging importlib_metadata
# install torch-hub specific dependencies
pip install -e git+https://github.com/huggingface/transformers.git#egg=transformers[torchhub]
# no longer needed
pip uninstall -y transformers
- name: Torch hub list
run: |

View File

@@ -229,6 +229,8 @@ extras["dev"] = (
+ extras["modelcreation"]
)
extras["torchhub"] = deps_list("filelock", "importlib_metadata", "numpy", "packaging", "protobuf", "regex",
"requests", "sacremoses", "sentencepiece", "torch", "tokenizers", "tqdm")
# when modifying the following list, make sure to update src/transformers/dependency_versions_check.py
install_requires = [