|
14 | 14 | setup( |
15 | 15 | name='timm', |
16 | 16 | version=__version__, |
17 | | - description='(Unofficial) PyTorch Image Models', |
| 17 | + description='PyTorch Image Models', |
18 | 18 | long_description=long_description, |
19 | 19 | long_description_content_type='text/markdown', |
20 | | - url='https://github.com/rwightman/pytorch-image-models', |
| 20 | + url='https://github.com/huggingface/pytorch-image-models', |
21 | 21 | author='Ross Wightman', |
22 | | - |
| 22 | + |
23 | 23 | classifiers=[ |
24 | 24 | # How mature is this project? Common values are |
25 | 25 | # 3 - Alpha |
|
29 | 29 | 'Intended Audience :: Education', |
30 | 30 | 'Intended Audience :: Science/Research', |
31 | 31 | 'License :: OSI Approved :: Apache Software License', |
32 | | - 'Programming Language :: Python :: 3.6', |
33 | 32 | 'Programming Language :: Python :: 3.7', |
34 | 33 | 'Programming Language :: Python :: 3.8', |
35 | 34 | 'Programming Language :: Python :: 3.9', |
36 | 35 | 'Programming Language :: Python :: 3.10', |
| 36 | + 'Programming Language :: Python :: 3.11', |
37 | 37 | 'Topic :: Scientific/Engineering', |
38 | 38 | 'Topic :: Scientific/Engineering :: Artificial Intelligence', |
39 | 39 | 'Topic :: Software Development', |
|
45 | 45 | keywords='pytorch pretrained models efficientnet mobilenetv3 mnasnet resnet vision transformer vit', |
46 | 46 | packages=find_packages(exclude=['convert', 'tests', 'results']), |
47 | 47 | include_package_data=True, |
48 | | - install_requires=['torch >= 1.7', 'torchvision', 'pyyaml', 'huggingface_hub'], |
49 | | - python_requires='>=3.6', |
| 48 | + install_requires=['torch >= 1.7', 'torchvision', 'pyyaml', 'huggingface_hub', 'safetensors'], |
| 49 | + python_requires='>=3.7', |
50 | 50 | ) |
51 | 51 |
|
0 commit comments