Unverified Commit b2dfcc56 authored by Stas Bekman's avatar Stas Bekman Committed by GitHub
Browse files

use correct deps for torchhub (#9552)

parent eabad8fd
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: |
......
......@@ -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 = [
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment