diff --git a/setup.py b/setup.py index 882ed467..04f121d8 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=['torch >= 1.4', 'torchvision'], + install_requires=['huggingface_hub', 'torch >= 1.4', 'torchvision'], python_requires='>=3.6', )