Dia2-2B / pyproject.toml
NariLabs's picture
Upload folder using huggingface_hub
1315cad verified
raw
history blame
923 Bytes
[build-system]
requires = ["setuptools>=70.0", "wheel"]
build-backend = "setuptools.build_meta"
[project]
name = "dia2"
version = "0.1.0"
description = "Dia2 CUDA-only text-to-speech runtime"
readme = "README.md"
requires-python = ">=3.10"
authors = [{ name = "Dia Contributors" }]
license = { file = "LICENSE" }
dependencies = [
"torch>=2.8.0",
"numpy>=2.1.0,<3.0",
"transformers>=4.55.3",
"safetensors==0.5.3",
"huggingface-hub>=0.24.7",
"sphn>=0.2.0",
"soundfile>=0.12.1",
"whisper-timestamped>=1.14.2",
"gradio>=4.44.1",
]
[project.optional-dependencies]
dev = [
"ruff>=0.6.9",
"pyright>=1.1.385",
]
[tool.uv]
package = true
[tool.uv.sources]
torch = [
{ index = "pytorch-cu128", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
]
[[tool.uv.index]]
name = "pytorch-cu128"
url = "https://download.pytorch.org/whl/cu128"
explicit = true
[tool.setuptools]
packages = ["dia2"]