From 9792e06f189e369496bd5124181ea9acafd9712e Mon Sep 17 00:00:00 2001 From: doctorpangloss <2229300+doctorpangloss@users.noreply.github.com> Date: Mon, 6 Oct 2025 15:21:00 -0700 Subject: [PATCH] Investigate torchaudio installation --- pyproject.toml | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 7d5fd3fee..97413bf8b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -149,6 +149,7 @@ cu126 = [ "torchvision", "triton ; sys_platform == 'Linux'", "triton-windows ; sys_platform == 'win32'", +# "torchaudio", ] cu128 = [ @@ -156,11 +157,13 @@ cu128 = [ "torchvision", "triton ; sys_platform == 'Linux'", "triton-windows ; sys_platform == 'win32'", +# "torchaudio", ] rocm = [ "torch", "torchvision", +# "torchaudio", "pytorch-triton-rocm", ] @@ -173,6 +176,7 @@ rocm = [ mps = [ "torch", "torchvision", +# "torchaudio", ] attention = [ @@ -255,18 +259,15 @@ 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'" }, +# { 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"