From b2dfcc567b2391c602208790562f4096638fef17 Mon Sep 17 00:00:00 2001
From: Stas Bekman <stas00@users.noreply.github.com>
Date: Wed, 13 Jan 2021 05:02:53 -0800
Subject: [PATCH] use correct deps for torchhub (#9552)

---
 .github/workflows/github-torch-hub.yml | 8 +++++---
 setup.py                               | 2 ++
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/github-torch-hub.yml b/.github/workflows/github-torch-hub.yml
index 6c39ae238e..0fcf4d326b 100644
--- a/.github/workflows/github-torch-hub.yml
+++ b/.github/workflows/github-torch-hub.yml
@@ -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: |
diff --git a/setup.py b/setup.py
index e6a7d89b45..e4856da9ad 100644
--- a/setup.py
+++ b/setup.py
@@ -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 = [