-
Notifications
You must be signed in to change notification settings - Fork 663
/
setup.py
63 lines (52 loc) · 1.81 KB
/
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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
import os
import re
from setuptools import find_packages, setup
_deps = [
"torch",
"xformers",
"diffusers==0.24.0",
"transformers",
"accelerate",
"fire",
"omegaconf",
"cuda-python",
"onnx==1.15.0",
"onnxruntime==1.16.3",
"protobuf==3.20.2",
"colored",
"pywin32;sys_platform == 'win32'"
]
deps = {b: a for a, b in (re.findall(r"^(([^!=<>~]+)(?:[!=<>~].*)?$)", x)[0] for x in _deps)}
def deps_list(*pkgs):
return [deps[pkg] for pkg in pkgs]
extras = {}
extras["xformers"] = deps_list("xformers")
extras["torch"] = deps_list("torch", "accelerate")
extras["tensorrt"] = deps_list("protobuf", "cuda-python", "onnx", "onnxruntime", "colored")
extras["dev"] = extras["xformers"] + extras["torch"] + extras["tensorrt"]
install_requires = [
deps["fire"],
deps["omegaconf"],
deps["diffusers"],
deps["transformers"],
deps["accelerate"],
]
setup(
name="streamdiffusion",
version="0.1.1",
description="real-time interactive image generation pipeline",
long_description=open("README.md", "r", encoding="utf-8").read(),
long_description_content_type="text/markdown",
keywords="deep learning diffusion pytorch stable diffusion audioldm streamdiffusion real-time",
license="Apache 2.0 License",
author="Aki, kizamimi, ddPn08, Verb, ramune, teftef6220, Tonimono, Chenfeng Xu, Ararat with the help of all our contributors (https://github.com/cumulo-autumn/StreamDiffusion/graphs/contributors)",
author_email="[email protected]",
url="https://github.com/cumulo-autumn/StreamDiffusion",
package_dir={"": "src"},
packages=find_packages("src"),
package_data={"streamdiffusion": ["py.typed"]},
include_package_data=True,
python_requires=">=3.10.0",
install_requires=list(install_requires),
extras_require=extras,
)