From c1b173e62a77b0c88143cd77470375e771aa140d Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Fri, 31 Jan 2025 09:13:47 -0800
Subject: [PATCH] Update to cu124 for torch 2.6.0 support

---
 README.md | 2 +-
 setup.py  | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index 0a37661a9..42744156e 100644
--- a/README.md
+++ b/README.md
@@ -156,7 +156,7 @@ When using Windows, open the **Windows Powershell** app. Then observe you are at
 
    To enable `torchaudio` support on Windows, install it directly:
    ```powershell
-   uv pip install torchaudio==2.5.1+cu121 --index-url https://download.pytorch.org/whl/cu121
+   uv pip install torchaudio==2.6.0+cu124 --index-url https://download.pytorch.org/whl/cu124
    ```
 7. To run the web server:
    ```shell
diff --git a/setup.py b/setup.py
index 77df620a1..695d43baa 100644
--- a/setup.py
+++ b/setup.py
@@ -28,13 +28,13 @@ version = "0.3.11"
 """
 The package index to the torch built with AMD ROCm.
 """
-amd_torch_index = ("https://download.pytorch.org/whl/rocm6.0", "https://download.pytorch.org/whl/nightly/rocm6.1")
+amd_torch_index = ("https://download.pytorch.org/whl/rocm6.2", "https://download.pytorch.org/whl/nightly/rocm6.2.4")
 
 """
 The package index to torch built with CUDA.
 Observe the CUDA version is in this URL.
 """
-nvidia_torch_index = ("https://download.pytorch.org/whl/cu121", "https://download.pytorch.org/whl/nightly/cu124")
+nvidia_torch_index = ("https://download.pytorch.org/whl/cu124", "https://download.pytorch.org/whl/nightly/cu126")
 
 """
 The package index to torch built against CPU features.