mrfakename commited on
Commit
61085f9
1 Parent(s): dacb26e

Sync from GitHub repo

Browse files

This Space is synced from the GitHub repo: https://github.com/SWivid/F5-TTS. Please submit contributions to the Space there

Files changed (1) hide show
  1. README_REPO.md +4 -1
README_REPO.md CHANGED
@@ -26,8 +26,11 @@
26
  conda create -n f5-tts python=3.10
27
  conda activate f5-tts
28
 
29
- # Install pytorch with your CUDA version, e.g.
30
  pip install torch==2.3.0+cu118 torchaudio==2.3.0+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
 
 
 
31
  ```
32
 
33
  Then you can choose from a few options below:
 
26
  conda create -n f5-tts python=3.10
27
  conda activate f5-tts
28
 
29
+ # NVIDIA GPU: install pytorch with your CUDA version, e.g.
30
  pip install torch==2.3.0+cu118 torchaudio==2.3.0+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
31
+
32
+ # AMD GPU: install pytorch with your ROCm version, e.g.
33
+ pip install torch==2.5.1+rocm6.2 torchaudio==2.5.1+rocm6.2 --extra-index-url https://download.pytorch.org/whl/rocm6.2
34
  ```
35
 
36
  Then you can choose from a few options below: