wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-10 06:10:50 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

Fix av dependency compatibility - too early to use 16, fix rocm libs, add torchaudio sources

Browse Source
This commit is contained in:
doctorpangloss 2025-10-06 11:32:17 -07:00
parent dcba4847ab
commit 4e56560f05
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
pyproject.toml
View File

@ -97,7 +97,7 @@ dependencies = [
"flax",
"jax",
"colour",
"av>=14.2.0",
"av>=14.2.0,<16",
"pydantic~=2.0",
"pydantic-settings~=2.0",
"typer",
@ -162,9 +162,14 @@ rocm = [
"torch",
"torchvision",
"pytorch-triton-rocm",
"rocm[libraries,devel]",
]
# todo: wip for rocm libraries with Windows and better RDNA support when installing with uv
# you'll need to look at the github actions workflow to see how this is done for e.g. an RX 7600
# therock = [
# "rocm[libraries,devel] ; sys_platform != 'Darwin'",
# ]
mps = [
"torch",
"torchvision",
@ -250,11 +255,18 @@ torchvision = [
]
torchaudio = [
{ index = "pytorch-cpu", extra = "cpu" },
{ index = "pytorch-cu126", extra = "cu126", marker = "sys_platform == 'Linux' or sys_platform == 'win32'" },
{ index = "pytorch-cu128", extra = "cu128", marker = "sys_platform == 'Linux' or sys_platform == 'win32'" },
{ index = "pytorch-rocm", extra = "rocm", marker = "sys_platform == 'Linux'" },
{ index = "pytorch-nightly-cpu", extra = "mps", marker = "sys_platform == 'darwin'" },
]
sageattention = [
{ git = "https://github.com/thu-ml/SageAttention.git", marker = "sys_platform == 'Linux'", extra = "attention" },
{ url = "https://github.com/woct0rdho/SageAttention/releases/download/v2.2.0-windows.post1/sageattention-2.2.0+cu128torch2.7.1.post1-cp39-abi3-win_amd64.whl", marker = "sys_platform == 'win32'", extra = "attention" },
]
audiocraft = [
{ git = "https://github.com/hiddenswitch/audiocraft-installable.git", extra = "music"}
]
[[tool.uv.dependency-metadata]]
name = "flash-attn"