Skip to content

Commit 11f7b58

Browse files
committed
Update setup.py for huggingface changes.
1 parent 4d9c3ae commit 11f7b58

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

setup.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -14,12 +14,12 @@
1414
setup(
1515
name='timm',
1616
version=__version__,
17-
description='(Unofficial) PyTorch Image Models',
17+
description='PyTorch Image Models',
1818
long_description=long_description,
1919
long_description_content_type='text/markdown',
20-
url='https://github.com/rwightman/pytorch-image-models',
20+
url='https://github.com/huggingface/pytorch-image-models',
2121
author='Ross Wightman',
22-
author_email='[email protected]',
22+
author_email='[email protected]',
2323
classifiers=[
2424
# How mature is this project? Common values are
2525
# 3 - Alpha
@@ -29,11 +29,11 @@
2929
'Intended Audience :: Education',
3030
'Intended Audience :: Science/Research',
3131
'License :: OSI Approved :: Apache Software License',
32-
'Programming Language :: Python :: 3.6',
3332
'Programming Language :: Python :: 3.7',
3433
'Programming Language :: Python :: 3.8',
3534
'Programming Language :: Python :: 3.9',
3635
'Programming Language :: Python :: 3.10',
36+
'Programming Language :: Python :: 3.11',
3737
'Topic :: Scientific/Engineering',
3838
'Topic :: Scientific/Engineering :: Artificial Intelligence',
3939
'Topic :: Software Development',
@@ -45,7 +45,7 @@
4545
keywords='pytorch pretrained models efficientnet mobilenetv3 mnasnet resnet vision transformer vit',
4646
packages=find_packages(exclude=['convert', 'tests', 'results']),
4747
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',
5050
)
5151

0 commit comments

Comments
 (0)