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', )