sin30 commited on
Commit
d127b6e
·
verified ·
1 Parent(s): 21a1823

Upload pyproject.toml

Browse files
Files changed (1) hide show
  1. pyproject.toml +65 -0
pyproject.toml ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [build-system]
2
+ requires = ["setuptools>=61.0", "wheel"]
3
+ build-backend = "setuptools.build_meta"
4
+
5
+ [project]
6
+ name = "xfuser"
7
+ dynamic = ["version"]
8
+ description = "A Scalable Inference Engine for Diffusion Transformers (DiTs) on Multiple Computing Devices"
9
+ readme = "README.md"
10
+ requires-python = ">=3.10"
11
+ license = {text = "MIT"}
12
+ authors = [
13
+ {name = "xDiT Team", email = "fangjiarui123@gmail.com"}
14
+ ]
15
+ classifiers = [
16
+ "Programming Language :: Python :: 3",
17
+ "Operating System :: OS Independent",
18
+ ]
19
+ dependencies = [
20
+ "torch>=2.1.0",
21
+ "accelerate>=0.33.0",
22
+ "transformers>=4.39.1",
23
+ "sentencepiece>=0.1.99",
24
+ "beautifulsoup4>=4.12.3",
25
+ "distvae",
26
+ "yunchang>=0.6.0",
27
+ "einops",
28
+ "diffusers>=0.35.1", # Need to update the diffusers version when supporting a new model.
29
+ "peft==0.17.0"
30
+ ]
31
+
32
+ [project.optional-dependencies]
33
+ flash-attn = [
34
+ "flash-attn>=2.6.0", # NOTE: flash-attn is necessary if ring_degree > 1
35
+ ]
36
+ optimum-quanto = [
37
+ "optimum-quanto", # NOTE: optimum-quanto is necessary if use_fp8_t5_encoder is enabled
38
+ ]
39
+ flask = [
40
+ "flask", # NOTE: flask is necessary to run xDiT as an http service
41
+ ]
42
+ ray = [
43
+ "ray", # NOTE: ray is necessary if RayDiffusionPipeline is used
44
+ ]
45
+ opencv-python = [
46
+ "opencv-python-headless", # NOTE: opencv-python is necessary if ConsisIDPipeline is used
47
+ ]
48
+ test = [
49
+ "pytest",
50
+ "imageio",
51
+ "imageio-ffmpeg"
52
+ ]
53
+
54
+ [project.urls]
55
+ Homepage = "https://github.com/xdit-project/xDiT."
56
+ Repository = "https://github.com/xdit-project/xDiT."
57
+
58
+ [tool.setuptools.dynamic]
59
+ version = {attr = "xfuser.__version__.__version__"}
60
+
61
+ [tool.setuptools.packages.find]
62
+ include = ["xfuser*"]
63
+
64
+ [tool.setuptools]
65
+ include-package-data = true