diff --git a/README.md b/README.md index a7c88cb2..e4d99dca 100644 --- 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 new file mode 100644 index 00000000..8b8d1122 --- /dev/null +++ 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 \ No newline at end of file