From aa6daf20af74a9acab1e81a62726fa9b6226cb2d Mon Sep 17 00:00:00 2001
From: Sylvain Gugger <sylvain.gugger@gmail.com>
Date: Thu, 18 Feb 2021 09:20:01 -0500
Subject: [PATCH] Add dependency everywhere

---
 requirements.txt | 1 +
 setup.py         | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index 2d29a27c..96834d83 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,4 @@
+huggingface-hub
 torch>=1.4.0
 torchvision>=0.5.0
 pyyaml
diff --git a/setup.py b/setup.py
index 04f121d8..4e3a831a 100644
--- a/setup.py
+++ b/setup.py
@@ -43,6 +43,6 @@ setup(
     keywords='pytorch pretrained models efficientnet mobilenetv3 mnasnet',
     packages=find_packages(exclude=['convert', 'tests', 'results']),
     include_package_data=True,
-    install_requires=['huggingface_hub', 'torch >= 1.4', 'torchvision'],
+    install_requires=['huggingface-hub', 'torch >= 1.4', 'torchvision'],
     python_requires='>=3.6',
 )