From 4e56560f056335096cbf1e7f583a5128a58edcb3 Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Mon, 6 Oct 2025 11:32:17 -0700
Subject: [PATCH] Fix av dependency compatibility - too early to use 16, fix
 rocm libs, add torchaudio sources

---
 pyproject.toml | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index b8f94cb2a..7d5fd3fee 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -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"