Provide instructions for AMD GPU

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)