andras/perfect-postcode
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 2 Wiki Activity Actions

Make vide work with prod

Browse source
This commit is contained in:
Andras Schmelczer 2026-05-10 14:55:53 +01:00
parent 3debacab4f
commit ee231d2ee5
7 changed files with 197 additions and 67 deletions
Download patch file Download diff file Expand all files Collapse all files

12
video/tts/pyproject.toml
View file

@ -16,6 +16,18 @@ dependencies = [
"numpy>=1.26",
]
[project.optional-dependencies]
# Flash-attention prebuilt wheel matched to torch 2.6 + cu12 + cp312, old
# CXX ABI (PyTorch's cu124 wheel reports compiled_with_cxx11_abi() == False
# and only exports the old-ABI c10::Error constructor). Pinned to
# 2.7.4.post1 because 2.8.x's torch2.6/abiFALSE wheels were mislabelled —
# they ship new-ABI symbols and fail to import. Building from source needs
# nvcc which isn't on the host. Enable via `uv sync --extra gpu`; render.sh
# does this automatically when nvidia-smi reports a GPU.
gpu = [
"flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1%2Bcu12torch2.6cxx11abiFALSE-cp312-cp312-linux_x86_64.whl ; sys_platform == 'linux' and platform_machine == 'x86_64'",
]
[tool.uv]
environments = ["sys_platform == 'linux' and python_version < '3.13'"]

25
video/tts/uv.lock generated
View file

@ -180,6 +180,24 @@ wheels = [
{ url = "https://files.pythonhosted.org/packages/81/47/dd9a212ef6e343a6857485ffe25bba537304f1913bdbed446a23f7f592e1/filelock-3.29.0-py3-none-any.whl", hash = "sha256:96f5f6344709aa1572bbf631c640e4ebeeb519e08da902c39a001882f30ac258", size = 39812, upload-time = "2026-04-19T15:39:08.752Z" },
]
[[package]]
name = "flash-attn"
version = "2.7.4.post1"
source = { url = "https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1%2Bcu12torch2.6cxx11abiFALSE-cp312-cp312-linux_x86_64.whl" }
dependencies = [
{ name = "einops", marker = "sys_platform == 'linux'" },
{ name = "torch", marker = "sys_platform == 'linux'" },
]
wheels = [
{ url = "https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1%2Bcu12torch2.6cxx11abiFALSE-cp312-cp312-linux_x86_64.whl", hash = "sha256:7e0b07913d56782d0f3f2ee76bd39587557742628983f6ab9ec2527c99437476" },
]
[package.metadata]
requires-dist = [
{ name = "einops" },
{ name = "torch" },
]
[[package]]
name = "flatbuffers"
version = "25.12.19"
@ -733,14 +751,21 @@ dependencies = [
{ name = "torchaudio", marker = "sys_platform == 'linux'" },
]
[package.optional-dependencies]
gpu = [
{ name = "flash-attn", marker = "platform_machine == 'x86_64' and sys_platform == 'linux'" },
]
[package.metadata]
requires-dist = [
{ name = "flash-attn", marker = "platform_machine == 'x86_64' and sys_platform == 'linux' and extra == 'gpu'", url = "https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1%2Bcu12torch2.6cxx11abiFALSE-cp312-cp312-linux_x86_64.whl" },
{ name = "numpy", specifier = ">=1.26" },
{ name = "qwen-tts", specifier = ">=0.1.1" },
{ name = "soundfile", specifier = ">=0.12" },
{ name = "torch", specifier = ">=2.5,<2.7", index = "https://download.pytorch.org/whl/cu124" },
{ name = "torchaudio", specifier = ">=2.5,<2.7", index = "https://download.pytorch.org/whl/cu124" },
]
provides-extras = ["gpu"]
[[package]]
name = "protobuf"