DiffSynth-Studio/pyproject.toml at main · modelscope/DiffSynth-Studio · GitHub

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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
[build-system]
requires = ["setuptools>=77", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "diffsynth"
dynamic = ["version"]
description = "Enjoy the magic of Diffusion models!"
readme = "README.md"
authors = [{name = "ModelScope Team"}]
license = "Apache-2.0"
license-files = ["LICENSE"]
requires-python = ">=3.10.1"
dependencies = [
    "torch>=2.0.0",
    "torchvision",
    "transformers",
    "imageio[ffmpeg]",
    "safetensors",
    "einops",
    "modelscope",
    "ftfy",
    "pandas",
    "accelerate",
    "peft",
]
classifiers = [
    "Programming Language :: Python :: 3",
    "Operating System :: OS Independent",
]

[tool.setuptools.packages.find]
where = ["./"]
include = ["diffsynth", "diffsynth.*"]

[tool.setuptools.dynamic]
version = {attr = "diffsynth.version.__version__"}

[project.optional-dependencies]
audio = [
    "av",
    "torchaudio",
    "torchcodec",
    "librosa"
]
quant = [
    "bitsandbytes",
    "comfy-kitchen",
    "entropack[cuda13]",
    "torchao>=0.16"
]
training = [
    "deepspeed"
]
logger = [
    "tensorboard",
    "swanlab",
    "wandb"
]
npu = [
    "torch==2.7.1+cpu",
    "torch-npu==2.7.1",
    "torchvision==0.22.1+cpu"
]
npu_aarch64 = [
    "torch==2.7.1",
    "torch-npu==2.7.1",
    "torchvision==0.22.1"
]
controlnet = [
    "controlnet_aux"
]
infiniteyou = [
    "insightface",
    "facexlib"
]
ses = [
    "PyWavelets"
]
nexusgen = [
    "qwen_vl_utils",
    "transformers==4.49.0"
]
yue2 = [
    "tiktoken"
]
all = [
    "av",
    "torchaudio",
    "torchcodec",
    "librosa",
    "bitsandbytes",
    "comfy-kitchen",
    "entropack[cuda13]",
    "torchao>=0.16",
    "deepspeed",
    "tensorboard",
    "swanlab",
    "wandb"
]

[tool.setuptools]
include-package-data = true

# uv resolves all extras together, so the CPU-only torch pinned by the `npu` extra needs the PyTorch CPU index.
# pip users pass `--extra-index-url https://download.pytorch.org/whl/cpu` instead (see docs/*/Pipeline_Usage/Setup.md).
[[tool.uv.index]]
name = "pytorch-cpu"
url = "https://download.pytorch.org/whl/cpu"
explicit = true

[tool.uv.sources]
torch = [{ index = "pytorch-cpu", extra = "npu" }]
torchvision = [{ index = "pytorch-cpu", extra = "npu" }]

[tool.uv]
# The NPU extras pin torch 2.7.1, while `quant` (and `all`) need torch>=2.10 through entropack.
conflicts = [
    [{ extra = "npu" }, { extra = "npu_aarch64" }, { extra = "quant" }],
    [{ extra = "npu" }, { extra = "npu_aarch64" }, { extra = "all" }],
]