diff --git a/MANIFEST.in b/MANIFEST.in new file mode 100644 index 00000000..4f2d1584 --- /dev/null +++ b/MANIFEST.in @@ -0,0 +1,2 @@ +include timm/models/pruned/*.txt + diff --git a/setup.py b/setup.py index ce493e70..bc940810 100644 --- a/setup.py +++ b/setup.py @@ -41,6 +41,7 @@ setup( # Note that this is a string of words separated by whitespace, not a list. keywords='pytorch pretrained models efficientnet mobilenetv3 mnasnet', packages=find_packages(exclude=['convert']), + include_package_data=True, install_requires=['torch >= 1.0', 'torchvision'], python_requires='>=3.6', )