Merge pull request #1027 from coqui-ai/dev

v0.5.0

.github/ISSUE_TEMPLATE/bug_report.md

@@ -37,8 +37,8 @@ In the worse case provide steps to reproduce the behaviour.
 You can either run `TTS/bin/collect_env_info.py`
 
 ```bash
-wget https://raw.githubusercontent.com/coqui-ai/TTS/main/TTS/bin/collect_env_details.py
-python collect_env_details.py
+wget https://raw.githubusercontent.com/coqui-ai/TTS/main/TTS/bin/collect_env_info.py
+python collect_env_info.py
 ```
 
 or fill in the fields below manually.

.github/workflows/aux_tests.yml

@@ -22,25 +22,22 @@ jobs:
         experimental: [false]
     steps:
       - uses: actions/checkout@v2
-      - uses: actions/cache@v1
-        with:
-          path: ~/.cache/pip
-          key: ${{ runner.os }}-pip-${{ matrix.python-version }}-${{ hashFiles('**/setup.py') }}
       - name: Set up Python ${{ matrix.python-version }}
-        uses: actions/setup-python@v2
+        uses: coqui-ai/setup-python@pip-cache-key-py-ver
         with:
           python-version: ${{ matrix.python-version }}
           architecture: x64
+          cache: 'pip'
+          cache-dependency-path: 'requirements*'
       - name: check OS
         run: cat /etc/os-release
       - name: Install dependencies
         run: |
-          sudo apt update
-          sudo apt install -y git make
-          sudo apt install -y python3-wheel gcc
+          sudo apt-get update
+          sudo apt-get install -y git make gcc
           make system-deps
-      - name: Upgrade pip
-        run: python3 -m pip install --upgrade pip
+      - name: Install/upgrade Python setup deps
+        run: python3 -m pip install --upgrade pip setuptools wheel
       - name: Install TTS
         run: |
           python3 -m pip install .[all]

.github/workflows/pypi-release.yml

@@ -7,7 +7,7 @@ defaults:
     shell:
       bash
 jobs:
-  build-package:
+  build-sdist:
     runs-on: ubuntu-20.04
     steps:
       - uses: actions/checkout@v2
@@ -23,16 +23,74 @@ jobs:
         with:
           python-version: 3.8
       - run: |
-          python -m pip install -U pip setuptools twine toml
-          python -c 'import toml; c = toml.load("pyproject.toml"); print("\n".join(c["build-system"]["requires"]))' | pip install -r /dev/stdin
+          python -m pip install -U pip setuptools wheel build
       - run: |
-          python setup.py sdist
+          python -m build
+      - run: |
+          pip install dist/*.tar.gz
+      - uses: actions/upload-artifact@v2
+        with:
+          name: sdist
+          path: dist/*.tar.gz
+  build-wheels:
+    runs-on: ubuntu-20.04
+    strategy:
+      matrix:
+        python-version: ["3.6", "3.7", "3.8", "3.9"]
+    steps:
+      - uses: actions/checkout@v2
+      - uses: actions/setup-python@v2
+        with:
+          python-version: ${{ matrix.python-version }}
+      - run: |
+          python -m pip install -U pip setuptools wheel build
+      - run: |
+          python -m build
+      - run: |
+          python -m pip install dist/*.whl
+      - uses: actions/upload-artifact@v2
+        with:
+          name: wheel-${{ matrix.python-version }}
+          path: dist/*.whl
+  publish-artifacts:
+    runs-on: ubuntu-20.04
+    needs: [build-sdist, build-wheels]
+    steps:
+      - run: |
+          mkdir dist
+      - uses: actions/download-artifact@v2
+        with:
+          name: "sdist"
+          path: "dist/"
+      - uses: actions/download-artifact@v2
+        with:
+          name: "wheel-3.6"
+          path: "dist/"
+      - uses: actions/download-artifact@v2
+        with:
+          name: "wheel-3.7"
+          path: "dist/"
+      - uses: actions/download-artifact@v2
+        with:
+          name: "wheel-3.8"
+          path: "dist/"
+      - uses: actions/download-artifact@v2
+        with:
+          name: "wheel-3.9"
+          path: "dist/"
+      - run: |
+          ls -lh dist/
       - name: Setup PyPI config
         run: |
           cat << EOF > ~/.pypirc
           [pypi]
           username=__token__
           password=${{ secrets.PYPI_TOKEN }}
           EOF
+      - uses: actions/setup-python@v2
+        with:
+          python-version: 3.8
+      - run: |
+          python -m pip install twine
       - run: |
-          twine upload --repository pypi dist/*.tar.gz
+          twine upload --repository pypi dist/*

.github/workflows/style_check.yml

@@ -22,25 +22,22 @@ jobs:
         experimental: [false]
     steps:
       - uses: actions/checkout@v2
-      - uses: actions/cache@v1
-        with:
-          path: ~/.cache/pip
-          key: ${{ runner.os }}-pip-${{ matrix.python-version }}-${{ hashFiles('**/setup.py') }}
       - name: Set up Python ${{ matrix.python-version }}
-        uses: actions/setup-python@v2
+        uses: coqui-ai/setup-python@pip-cache-key-py-ver
         with:
           python-version: ${{ matrix.python-version }}
           architecture: x64
+          cache: 'pip'
+          cache-dependency-path: 'requirements*'
       - name: check OS
         run: cat /etc/os-release
       - name: Install dependencies
         run: |
-          sudo apt update
-          sudo apt install -y git make
-          sudo apt install -y python3-wheel gcc
+          sudo apt-get update
+          sudo apt-get install -y git make gcc
           make system-deps
-      - name: Upgrade pip
-        run: python3 -m pip install --upgrade pip
+      - name: Install/upgrade Python setup deps
+        run: python3 -m pip install --upgrade pip setuptools wheel
       - name: Install TTS
         run: |
           python3 -m pip install .[all]

.github/workflows/tts_tests.yml

@@ -22,25 +22,22 @@ jobs:
         experimental: [false]
     steps:
       - uses: actions/checkout@v2
-      - uses: actions/cache@v1
-        with:
-          path: ~/.cache/pip
-          key: ${{ runner.os }}-pip-${{ matrix.python-version }}-${{ hashFiles('**/setup.py') }}
       - name: Set up Python ${{ matrix.python-version }}
-        uses: actions/setup-python@v2
+        uses: coqui-ai/setup-python@pip-cache-key-py-ver
         with:
           python-version: ${{ matrix.python-version }}
           architecture: x64
+          cache: 'pip'
+          cache-dependency-path: 'requirements*'
       - name: check OS
         run: cat /etc/os-release
       - name: Install dependencies
         run: |
-          sudo apt update
-          sudo apt install -y git make
-          sudo apt install -y python3-wheel gcc
+          sudo apt-get update
+          sudo apt-get install -y --no-install-recommends git make gcc
           make system-deps
-      - name: Upgrade pip
-        run: python3 -m pip install --upgrade pip
+      - name: Install/upgrade Python setup deps
+        run: python3 -m pip install --upgrade pip setuptools wheel
       - name: Install TTS
         run: |
           python3 -m pip install .[all]

.github/workflows/vocoder_tests.yml

@@ -22,25 +22,22 @@ jobs:
         experimental: [false]
     steps:
       - uses: actions/checkout@v2
-      - uses: actions/cache@v1
-        with:
-          path: ~/.cache/pip
-          key: ${{ runner.os }}-pip-${{ matrix.python-version }}-${{ hashFiles('**/setup.py') }}
       - name: Set up Python ${{ matrix.python-version }}
-        uses: actions/setup-python@v2
+        uses: coqui-ai/setup-python@pip-cache-key-py-ver
         with:
           python-version: ${{ matrix.python-version }}
           architecture: x64
+          cache: 'pip'
+          cache-dependency-path: 'requirements*'
       - name: check OS
         run: cat /etc/os-release
       - name: Install dependencies
         run: |
-          sudo apt update
-          sudo apt install -y git make
-          sudo apt install -y python3-wheel gcc
+          sudo apt-get update
+          sudo apt-get install -y git make gcc
           make system-deps
-      - name: Upgrade pip
-        run: python3 -m pip install --upgrade pip
+      - name: Install/upgrade Python setup deps
+        run: python3 -m pip install --upgrade pip setuptools wheel
       - name: Install TTS
         run: |
           python3 -m pip install .[all]

.github/workflows/zoo_tests.yml

@@ -22,25 +22,22 @@ jobs:
         experimental: [false]
     steps:
       - uses: actions/checkout@v2
-      - uses: actions/cache@v1
-        with:
-          path: ~/.cache/pip
-          key: ${{ runner.os }}-pip-${{ matrix.python-version }}-${{ hashFiles('**/setup.py') }}
       - name: Set up Python ${{ matrix.python-version }}
-        uses: actions/setup-python@v2
+        uses: coqui-ai/setup-python@pip-cache-key-py-ver
         with:
           python-version: ${{ matrix.python-version }}
           architecture: x64
+          cache: 'pip'
+          cache-dependency-path: 'requirements*'
       - name: check OS
         run: cat /etc/os-release
       - name: Install dependencies
         run: |
-          sudo apt update
-          sudo apt install -y git make
-          sudo apt install -y python3-wheel gcc
+          sudo apt-get update
+          sudo apt-get install -y git make gcc
           make system-deps
-      - name: Upgrade pip
-        run: python3 -m pip install --upgrade pip
+      - name: Install/upgrade Python setup deps
+        run: python3 -m pip install --upgrade pip setuptools wheel
       - name: Install TTS
         run: |
           python3 -m pip install .[all]

.gitignore

@@ -128,6 +128,8 @@ core
 recipes/WIP/*
 recipes/ljspeech/LJSpeech-1.1/*
 recipes/vctk/VCTK/*
+recipes/**/*.npy
+recipes/**/*.json
 VCTK-Corpus-removed-silence/*
 
 # ignore training logs
@@ -161,4 +163,5 @@ speakers.json
 internal/*
 *_pitch.npy
 *_phoneme.npy
-wandb
+wandb
+depot/*

README.md

@@ -1,7 +1,7 @@
 # <img src="https://raw.githubusercontent.com/coqui-ai/TTS/main/images/coqui-log-green-TTS.png" height="56"/>
 
 🐸TTS is a library for advanced Text-to-Speech generation. It's built on the latest research, was designed to achieve the best trade-off among ease-of-training, speed and quality.
-🐸TTS comes with [pretrained models](https://github.com/coqui-ai/TTS/wiki/Released-Models), tools for measuring dataset quality and already used in **20+ languages** for products and research projects.
+🐸TTS comes with pretrained models, tools for measuring dataset quality and already used in **20+ languages** for products and research projects.
 
 [![GithubActions](https://github.com/coqui-ai/TTS/actions/workflows/main.yml/badge.svg)](https://github.com/coqui-ai/TTS/actions)
 [![PyPI version](https://badge.fury.io/py/TTS.svg)](https://badge.fury.io/py/TTS)
@@ -135,6 +135,66 @@ $ make install
 
 If you are on Windows, 👑@GuyPaddock wrote installation instructions [here](https://stackoverflow.com/questions/66726331/how-can-i-run-mozilla-tts-coqui-tts-training-with-cuda-on-a-windows-system).
 
+## Use TTS
+
+### Single Speaker Models
+
+- List provided models:
+
+    ```
+    $ tts --list_models
+    ```
+
+- Run TTS with default models:
+
+    ```
+    $ tts --text "Text for TTS"
+    ```
+
+- Run a TTS model with its default vocoder model:
+
+    ```
+    $ tts --text "Text for TTS" --model_name "<language>/<dataset>/<model_name>
+    ```
+
+- Run with specific TTS and vocoder models from the list:
+
+    ```
+    $ tts --text "Text for TTS" --model_name "<language>/<dataset>/<model_name>" --vocoder_name "<language>/<dataset>/<model_name>" --output_path
+    ```
+
+- Run your own TTS model (Using Griffin-Lim Vocoder):
+
+    ```
+    $ tts --text "Text for TTS" --model_path path/to/model.pth.tar --config_path path/to/config.json --out_path output/path/speech.wav
+    ```
+
+- Run your own TTS and Vocoder models:
+    ```
+    $ tts --text "Text for TTS" --model_path path/to/config.json --config_path path/to/model.pth.tar --out_path output/path/speech.wav
+        --vocoder_path path/to/vocoder.pth.tar --vocoder_config_path path/to/vocoder_config.json
+    ```
+
+### Multi-speaker Models
+
+- List the available speakers and choose as <speaker_id> among them:
+
+    ```
+    $ tts --model_name "<language>/<dataset>/<model_name>"  --list_speaker_idxs
+    ```
+
+- Run the multi-speaker TTS model with the target speaker ID:
+
+    ```
+    $ tts --text "Text for TTS." --out_path output/path/speech.wav --model_name "<language>/<dataset>/<model_name>"  --speaker_idx <speaker_id>
+    ```
+
+- Run your own multi-speaker TTS model:
+
+    ```
+    $ tts --text "Text for TTS" --out_path output/path/speech.wav --model_path path/to/config.json --config_path path/to/model.pth.tar --speakers_file_path path/to/speaker.json --speaker_idx <speaker_id>
+    ```
+
 ## Directory Structure
 ```
 |- notebooks/       (Jupyter Notebooks for model evaluation, parameter selection and data analysis.)