aria/pyproject.toml at main · EleutherAI/aria · 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
[build-system]
requires = ["setuptools>=61.0"]
build-backend = "setuptools.build_meta"

[project]
name = "aria"
version = "0.0.1"
description = ""
authors = [{name = "Louis Bradshaw", email = "loua19@outlook.com"}]
requires-python = ">=3.11"

dependencies = [
    "ariautils @ git+https://github.com/EleutherAI/aria-utils.git",
    "torch>=2.3",
    "mlx<=0.26",
    "safetensors",
    "jsonlines",
    "tqdm",
]

[project.optional-dependencies]
dev = ["ruff", "ty"]
train = ["accelerate"]
eval = ["transformers", "torchaudio", "mido"]
demo = ["python-rtmidi", "numpy"]
all = ["black", "accelerate", "transformers", "torchaudio", "mido", "python-rtmidi", "numpy"]

[tool.black]
line-length = 80
target-version = ["py311"]
include = '\.pyi?$'

[project.scripts]
aria = "aria.run:main"

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

[tool.setuptools.package-data]
aria = ["../config/*.json", "../config/models/*.json"]