Fix pruned txt files not being installed during pip install

pull/140/head
Ross Wightman 5 years ago
parent 8ec554b82e
commit 375f3e5321

@ -0,0 +1,2 @@
include timm/models/pruned/*.txt

@ -41,6 +41,7 @@ setup(
# Note that this is a string of words separated by whitespace, not a list. # Note that this is a string of words separated by whitespace, not a list.
keywords='pytorch pretrained models efficientnet mobilenetv3 mnasnet', keywords='pytorch pretrained models efficientnet mobilenetv3 mnasnet',
packages=find_packages(exclude=['convert']), packages=find_packages(exclude=['convert']),
include_package_data=True,
install_requires=['torch >= 1.0', 'torchvision'], install_requires=['torch >= 1.0', 'torchvision'],
python_requires='>=3.6', python_requires='>=3.6',
) )

Loading…
Cancel
Save