From 7370ce72434d1fcb12f0c075c3e0fbff0bc341ee Mon Sep 17 00:00:00 2001
From: Rek-Malorm <70321198+Rek-Malorm@users.noreply.github.com>
Date: Fri, 27 Dec 2024 11:26:20 +1300
Subject: [PATCH 1/2] Provide instructions for AMD GPU

---
 README.md | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/README.md b/README.md
index c74e016..1d9e162 100644
--- a/README.md
+++ b/README.md
@@ -25,11 +25,22 @@
 # Create a python 3.10 conda env (you could also use virtualenv)
 conda create -n f5-tts python=3.10
 conda activate f5-tts
+```
 
+### Nvidia GPU
+
+```bash
 # Install pytorch with your CUDA version, e.g.
 pip install torch==2.3.0+cu118 torchaudio==2.3.0+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
 ```
 
+### AMD GPU
+
+```bash
+# Install pytorch with your ROCm version, e.g.
+pip install torch==2.5.1+rocm6.2 torchaudio==2.5.1+rocm6.2 --extra-index-url https://download.pytorch.org/whl/rocm6.2
+```
+
 Then you can choose from a few options below:
 
 ### 1. As a pip package (if just for inference)

From 6281b4b8edd7fd7e8741a4e3282afefc40174fe4 Mon Sep 17 00:00:00 2001
From: Yushen CHEN <45333109+SWivid@users.noreply.github.com>
Date: Fri, 27 Dec 2024 15:21:22 +0800
Subject: [PATCH 2/2] Update README.md

---
 README.md | 12 ++----------
 1 file changed, 2 insertions(+), 10 deletions(-)

diff --git a/README.md b/README.md
index 1d9e162..3e56efb 100644
--- a/README.md
+++ b/README.md
@@ -25,19 +25,11 @@
 # Create a python 3.10 conda env (you could also use virtualenv)
 conda create -n f5-tts python=3.10
 conda activate f5-tts
-```
 
-### Nvidia GPU
-
-```bash
-# Install pytorch with your CUDA version, e.g.
+# NVIDIA GPU: install pytorch with your CUDA version, e.g.
 pip install torch==2.3.0+cu118 torchaudio==2.3.0+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
-```
 
-### AMD GPU
-
-```bash
-# Install pytorch with your ROCm version, e.g.
+# AMD GPU: install pytorch with your ROCm version, e.g.
 pip install torch==2.5.1+rocm6.2 torchaudio==2.5.1+rocm6.2 --extra-index-url https://download.pytorch.org/whl/rocm6.2
 ```