Add dependency everywhere

pull/440/head
Sylvain Gugger 4 years ago
parent 614b73bd35
commit aa6daf20af

@ -1,3 +1,4 @@
huggingface-hub
torch>=1.4.0
torchvision>=0.5.0
pyyaml

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

Loading…
Cancel
Save