[tool.poetry]
name = "beatrice-trainer"
version = "2.0.0b2"
description = "A tool to train Beatrice models"
license = "MIT"
authors = ["Project Beatrice <167534685+prj-beatrice@users.noreply.github.com>"]
readme = "README.md"
homepage = "https://prj-beatrice.com/"
repository = "https://huggingface.co/fierce-cats/beatrice-trainer"

[tool.poetry.dependencies]
python = ">=3.9"
torch = [
    { version = ">=2.1", markers = "sys_platform == 'win32'", source = "torch-cuda" },
    { version = ">=2.1", markers = "sys_platform != 'win32'" },
]
torchaudio = [
    { version = ">=2.1", markers = "sys_platform == 'win32'", source = "torch-cuda" },
    { version = ">=2.1", markers = "sys_platform != 'win32'" },
]
tqdm = ">=4"
numpy = "^1"
tensorboard = ">=2"
soundfile = ">=0.11"
pyworld = ">=0.3.2"

[[tool.poetry.source]]
name = "torch-cuda"
url = "https://download.pytorch.org/whl/cu121"
priority = "explicit"

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"