diff --git a/setup.py b/setup.py index 2deffe9..42e12e3 100644 --- a/setup.py +++ b/setup.py @@ -17,7 +17,7 @@ setup( license="MIT", url="https://github.com/microsoft/torchscale", packages=find_packages(exclude=["*.tests", "*.tests.*", "tests.*", "tests"]), - install_requires=["torch>=1.8", "fairscale==0.4.0", "timm==0.4.12"], + install_requires=["torch>=1.8", "fairscale==0.4.0", "timm==0.6.13"], python_requires=">=3.8.0", classifiers=[ "Programming Language :: Python :: 3",