wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-22 12:20:16 +08:00
Code Issues Actions 15 Packages Projects Releases Wiki Activity

Investigate torchaudio installation

Browse Source
This commit is contained in:
doctorpangloss 2025-10-06 15:21:00 -07:00
parent 4e56560f05
commit 9792e06f18
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
pyproject.toml
View File

@ -149,6 +149,7 @@ cu126 = [
"torchvision", "torchvision",
"triton ; sys_platform == 'Linux'", "triton ; sys_platform == 'Linux'",
"triton-windows ; sys_platform == 'win32'", "triton-windows ; sys_platform == 'win32'",
# "torchaudio",
] ]
cu128 = [ cu128 = [
@ -156,11 +157,13 @@ cu128 = [
"torchvision", "torchvision",
"triton ; sys_platform == 'Linux'", "triton ; sys_platform == 'Linux'",
"triton-windows ; sys_platform == 'win32'", "triton-windows ; sys_platform == 'win32'",
# "torchaudio",
] ]
rocm = [ rocm = [
"torch", "torch",
"torchvision", "torchvision",
# "torchaudio",
"pytorch-triton-rocm", "pytorch-triton-rocm",
] ]
@ -173,6 +176,7 @@ rocm = [
mps = [ mps = [
"torch", "torch",
"torchvision", "torchvision",
# "torchaudio",
] ]
attention = [ attention = [
@ -255,18 +259,15 @@ torchvision = [
] ]
torchaudio = [ torchaudio = [
{ index = "pytorch-cpu", extra = "cpu" }, { index = "pytorch-cpu", extra = "cpu" },
{ index = "pytorch-cu126", extra = "cu126", marker = "sys_platform == 'Linux' or sys_platform == 'win32'" }, # { 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-cu128", extra = "cu128", marker = "sys_platform == 'Linux' or sys_platform == 'win32'" },
{ index = "pytorch-rocm", extra = "rocm", marker = "sys_platform == 'Linux'" }, # { index = "pytorch-rocm", extra = "rocm", marker = "sys_platform == 'Linux'" },
{ index = "pytorch-nightly-cpu", extra = "mps", marker = "sys_platform == 'darwin'" }, # { index = "pytorch-nightly-cpu", extra = "mps", marker = "sys_platform == 'darwin'" },
] ]
sageattention = [ sageattention = [
{ git = "https://github.com/thu-ml/SageAttention.git", marker = "sys_platform == 'Linux'", extra = "attention" }, { 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" }, { 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]] [[tool.uv.dependency-metadata]]
name = "flash-attn" name = "flash-attn"