mirror of
https://github.com/Nighthawk42/soprano-factory.git
synced 2026-08-30 20:42:26 +00:00
Nighthawk
9b7dd275c7
- Move hyperparameters from hardcoded script values to `config.yaml` - Replace pip requirements with `pyproject.toml` and `uv` support (CUDA 12.6) - Refactor all scripts to use `pathlib` for robust path handling - Optimize `generate_dataset.py` with GPU acceleration - Register quantizer constants as buffers for proper device mapping - Update README with new installation and usage instructions
35 lines
640 B
TOML
35 lines
640 B
TOML
[project]
|
|
name = "soprano-factory"
|
|
version = "0.1.0"
|
|
dependencies = [
|
|
"transformers",
|
|
"numpy",
|
|
"scipy",
|
|
"tqdm",
|
|
"einops",
|
|
"pyyaml",
|
|
"huggingface-hub",
|
|
"torch",
|
|
"torchaudio",
|
|
]
|
|
|
|
[project.optional-dependencies]
|
|
# Use uv sync --extra gpu to install CUDA support
|
|
gpu = [
|
|
"torch",
|
|
"torchaudio",
|
|
]
|
|
|
|
[[tool.uv.index]]
|
|
name = "pytorch-cu126"
|
|
url = "https://download.pytorch.org/whl/cu126"
|
|
explicit = true
|
|
|
|
[tool.uv.sources]
|
|
# Only use the CUDA index when the 'gpu' extra is requested
|
|
torch = [
|
|
{ index = "pytorch-cu126", extra = "gpu" },
|
|
]
|
|
torchaudio = [
|
|
{ index = "pytorch-cu126", extra = "gpu" },
|
|
] |