feat: requirements txt + windows

NewKcriss · Oct 4, 2023 · 4fd03b6 · 4fd03b6
1 parent 51063f3
commit 4fd03b6
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -23,7 +23,11 @@ cd StyleTTS2
 ```
 3. Install python requirements: 
 ```bash
-pip install SoundFile torchaudio munch torch pydub pyyaml librosa nltk matplotlib accelerate transformers einops einops-exts tqdm typing typing-extensions git+https://github.com/resemble-ai/monotonic_align.git
+pip install -r requirements.txt
+```
+On Windows add:
+```bash
+pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 -U
 ```
 Also install phonemizer and espeak if you want to run the demo:
 ```bash

diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,17 @@
+SoundFile
+torchaudio
+munch
+torch
+pydub
+pyyaml
+librosa
+nltk
+matplotlib
+accelerate
+transformers
+einops
+einops-exts
+tqdm
+typing
+typing-extensions
+git+https://github.com/resemble-ai/monotonic_align.git