Make vide work with prod
This commit is contained in:
parent
3debacab4f
commit
ee231d2ee5
7 changed files with 197 additions and 67 deletions
|
|
@ -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'"]
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue