-
Notifications
You must be signed in to change notification settings - Fork 252
/
Copy pathpyproject.toml
93 lines (85 loc) · 2.06 KB
/
pyproject.toml
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
[tool.poetry]
name = "ultravox"
version = "0.1.0"
description = ""
authors = ["Fixie.ai Team <[email protected]>"]
packages = [{ include = "ultravox/" }]
license = "MIT"
readme = "README.md"
[tool.poetry.dependencies]
python = ">=3.10,<4.0"
torch = ">=2.4"
transformers = {version = ">=4.43.1,<4.48.0 || >4.48.0", extras = ["torch"]}
huggingface-hub = ">=0.27.0"
peft = "~0.11.1"
simple-parsing = "~0.1.5"
librosa = "~0.10.2.post1"
requests = "~2.31.0"
datasets = "~2.19.1"
mosaicml-streaming = "~0.7.6"
nltk = "~3.8.1"
truecase = "~0.0.14"
sentencepiece = "~0.2.0"
protobuf = "~4.25.3"
dataclasses-json = "~0.6.7"
openai = ">=1.60.0"
jiwer = "~3.0.4"
tensorboardx = "~2.6.2.2"
wandb = "~0.17.1"
sacrebleu = {version = "^2.4.2"}
mecab-ko = "*" # Required for Korean tokenization in sacrebleu
mecab-python3 = "*" # Required for Japanese tokenization in sacrebleu
ipadic = "*" # IPA dictionary for MeCab
tenacity = "^9.0.0"
evals = {git = "https://github.com/fixie-ai/evals", rev = "2734b4318030cd15b8940b0761fff47fb0c1ad53"}
torchaudio = "^2.4.1"
scipy = "^1.14.1"
einops = "^0.8.0"
praatio = "^6.2.0"
hf-transfer = "^0.1.8"
annoy = "^1.17.3"
coverage = "^7.6.10"
whisper_normalizer = "^0.0.10"
evaluate = "^0.4.3"
[tool.poetry.group.dev.dependencies]
black = "~24.4.2"
isort = "~5.13.2"
mypy = "~1.10.0"
autoflake = "~2.3.1"
pytest = "~8.2.2"
fsspec = "~2024.3.1"
gcsfs = "~2024.3.1"
sounddevice = "~0.4.7"
mosaicml-cli = "~0.6.31"
gradio-webrtc = {version = ">0.0.19", extras = ["vad"]}
gpustat = "~1.1.1"
types-requests = "~2.26.0"
types-pyyaml = "^6.0.12.20240724"
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"
[tool.mypy]
python_version = "3.11"
warn_return_any = true
warn_unused_configs = true
[tool.isort]
profile = "black"
single_line_exclusions = ["typing", "collections.abc", "typing_extensions"]
skip = [
"venv",
".venv",
"third_party",
]
[tool.black]
extend-exclude = '''
/(
| third_party
| venv
)/
'''
[tool.coverage.run]
relative_files = true
source = ["ultravox"]
branch = true
[tool.coverage.report]
show_missing = true