-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
25 lines (20 loc) · 811 Bytes
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
import pathlib
import pkg_resources
from setuptools import find_packages, setup
with pathlib.Path("requirements.txt").open() as f:
install_requires = [str(requirement) for requirement in pkg_resources.parse_requirements(f)]
setup(
name="stabletriton",
version="0.1",
license="Apache License 2.0",
license_files=("LICENSE",),
description="Accelerate diffusion model inference with custom GPU kernels written in Triton",
long_description=open("README.md", "r", encoding="utf-8").read(),
keywords=["Deep Learning"],
long_description_content_type="text/markdown",
url="https://github.com/arnavdantuluri/StableTriton/tree/main",
package_dir={"": "src"},
packages=find_packages(where="src"),
install_requires=install_requires,
python_requires="==3.10.*",
)