pyproject.toml 510 Bytes
Newer Older
sxtyzhangzk's avatar
sxtyzhangzk committed
1
2
3
[build-system]
requires = [
    "setuptools",
4
    "torch>=2.5",
sxtyzhangzk's avatar
sxtyzhangzk committed
5
6
7
8
9
    "wheel",
    "ninja",
]
build-backend = "setuptools.build_meta"

10
11
12
[tool.setuptools.packages.find]
include = ["nunchaku"]

Hyunsung Lee's avatar
Hyunsung Lee committed
13
14
15
16
17
18
19
[tool.ruff]
line-length = 140

[tool.ruff.lint]
select = ["E", "W"]
ignore = ["F401", "E501"]

sxtyzhangzk's avatar
sxtyzhangzk committed
20
21
22
23
[project]
dynamic = ["version"]
name = "nunchaku"
dependencies = [
24
    "diffusers>=0.32.2",
sxtyzhangzk's avatar
sxtyzhangzk committed
25
26
27
28
    "transformers",
    "accelerate",
    "sentencepiece",
    "protobuf",
29
    "huggingface_hub",
30
]
muyangli's avatar
muyangli committed
31
requires-python = ">=3.10, <3.13"