torchscale/setup.py

26 lines
821 B
Python
Raw Permalink Normal View History

2022-11-23 16:36:55 +00:00
# Copyright (c) 2022 Microsoft
# Licensed under The MIT License [see LICENSE for details]
2022-11-23 16:21:58 +00:00
from io import open
2022-11-26 17:01:02 +00:00
2022-11-23 16:21:58 +00:00
from setuptools import find_packages, setup
setup(
name="torchscale",
2023-03-15 07:20:00 +00:00
version="0.2.0",
2022-11-23 16:21:58 +00:00
author="TorchScale Team",
author_email="Shuming.Ma@microsoft.com",
description="Transformers at any scale",
2022-11-26 17:01:02 +00:00
long_description=open("README.md", "r", encoding="utf-8").read(),
2022-11-23 16:21:58 +00:00
long_description_content_type="text/markdown",
keywords="Transformers at any scale",
license="MIT",
2023-03-04 09:11:34 +00:00
url="https://github.com/microsoft/torchscale",
2022-11-26 17:01:02 +00:00
packages=find_packages(exclude=["*.tests", "*.tests.*", "tests.*", "tests"]),
2023-04-11 10:02:21 +00:00
install_requires=["torch>=1.8", "fairscale==0.4.0", "timm==0.6.13"],
2022-11-26 17:01:02 +00:00
python_requires=">=3.8.0",
2022-11-23 16:21:58 +00:00
classifiers=[
2022-11-26 17:01:02 +00:00
"Programming Language :: Python :: 3",
2022-11-23 16:21:58 +00:00
],
2022-11-26 16:10:15 +00:00
)