diff --git a/README.md b/README.md index 0c78d8a..4e0c528 100644 --- a/README.md +++ b/README.md @@ -154,7 +154,13 @@ or python matcha/train.py experiment=ljspeech ``` -for multi-gpu training, run +- for a minimum memory run + +```bash +python matcha/train.py experiment=ljspeech_min_memory +``` + +- for multi-gpu training, run ```bash python matcha/train.py experiment=ljspeech trainer.devices=[0,1] @@ -162,10 +168,11 @@ python matcha/train.py experiment=ljspeech trainer.devices=[0,1] ## Acknowledgements -Since this code uses: [Lightning-Hydra-Template](https://github.com/ashleve/lightning-hydra-template), you have all the powers attached to it. +Since this code uses: [Lightning-Hydra-Template](https://github.com/ashleve/lightning-hydra-template), you have all the powers that comes with it. Other source codes I would like to acknowledge: -- [Coqui-TTS](https://github.com/coqui-ai/TTS/tree/dev) -- [Grad-TTS](https://github.com/huawei-noah/Speech-Backbones/tree/main/Grad-TTS) -- [torchdyn](https://github.com/DiffEqML/torchdyn) +- [Coqui-TTS](https://github.com/coqui-ai/TTS/tree/dev) : For helping me figure out how to make cython binaries pip installable +- [Grad-TTS](https://github.com/huawei-noah/Speech-Backbones/tree/main/Grad-TTS): For source code of MAS +- [torchdyn](https://github.com/DiffEqML/torchdyn): Useful for trying other ODE solvers during development +- [labml.ai](https://nn.labml.ai/transformers/rope/index.html): For RoPE implementation diff --git a/setup.py b/setup.py index a119858..516d410 100644 --- a/setup.py +++ b/setup.py @@ -2,7 +2,6 @@ import os import numpy -import pkg_resources from Cython.Build import cythonize from setuptools import Extension, find_packages, setup @@ -22,7 +21,7 @@ setup( url="https://shivammehta25.github.io/Matcha-TTS", install_requires=[ str(r) - for r in pkg_resources.parse_requirements(open(os.path.join(os.path.dirname(__file__), "requirements.txt"))) + for r in open(os.path.join(os.path.dirname(__file__), "requirements.txt")) ], include_dirs=[numpy.get_include()], packages=find_packages(exclude=["tests", "tests/*", "examples", "examples/*"]),