|
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