From 8af3d2dbcd89b2ef5915c2bab7eeb14833da67b2 Mon Sep 17 00:00:00 2001
From: Edresson Casanova <edresson1@gmail.com>
Date: Tue, 24 Oct 2023 09:52:44 -0300
Subject: [PATCH] Add a dedicated workflow for XTTS tests

---
 .github/workflows/xtts_tests.yml              | 53 +++++++++++++++++++
 Makefile                                      |  3 ++
 recipes/ljspeech/xtts_v1/train_gpt_xtts.py    |  1 +
 .../test_xtts_gpt_train.py                    |  0
 4 files changed, 57 insertions(+)
 create mode 100644 .github/workflows/xtts_tests.yml
 rename tests/{tts_tests2 => xtts_tests}/test_xtts_gpt_train.py (100%)

diff --git a/.github/workflows/xtts_tests.yml b/.github/workflows/xtts_tests.yml
new file mode 100644
index 0000000000..be367f3547
--- /dev/null
+++ b/.github/workflows/xtts_tests.yml
@@ -0,0 +1,53 @@
+name: xtts-tests
+
+on:
+  push:
+    branches:
+      - main
+  pull_request:
+    types: [opened, synchronize, reopened]
+jobs:
+  check_skip:
+    runs-on: ubuntu-latest
+    if: "! contains(github.event.head_commit.message, '[ci skip]')"
+    steps:
+      - run: echo "${{ github.event.head_commit.message }}"
+
+  test:
+    runs-on: ubuntu-latest
+    strategy:
+      fail-fast: false
+      matrix:
+        python-version: [3.9, "3.10", "3.11"]
+        experimental: [false]
+    steps:
+      - uses: actions/checkout@v3
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v4
+        with:
+          python-version: ${{ matrix.python-version }}
+          architecture: x64
+          cache: 'pip'
+          cache-dependency-path: 'requirements*'
+      - name: check OS
+        run: cat /etc/os-release
+      - name: set ENV
+        run: export TRAINER_TELEMETRY=0
+      - name: Install dependencies
+        run: |
+          sudo apt-get update
+          sudo apt-get install -y --no-install-recommends git make gcc
+          sudo apt-get install espeak
+          sudo apt-get install espeak-ng
+          make system-deps
+      - name: Install/upgrade Python setup deps
+        run: python3 -m pip install --upgrade pip setuptools wheel
+      - name: Replace scarf urls
+        run: |
+          sed -i 's/https:\/\/coqui.gateway.scarf.sh\//https:\/\/github.com\/coqui-ai\/TTS\/releases\/download\//g' TTS/.models.json
+      - name: Install TTS
+        run: |
+          python3 -m pip install .[all]
+          python3 setup.py egg_info
+      - name: Unit tests
+        run: make test_xtts
diff --git a/Makefile b/Makefile
index ab992ec52e..54aa6eeb18 100644
--- a/Makefile
+++ b/Makefile
@@ -22,6 +22,9 @@ test_tts:	## run tts tests.
 test_tts2:	## run tts tests.
 	nose2 -F -v -B --with-coverage --coverage TTS tests.tts_tests2
 
+test_xtts:
+	nose2 -F -v -B --with-coverage --coverage TTS tests.xtts_tests
+
 test_aux:	## run aux tests.
 	nose2 -F -v -B --with-coverage --coverage TTS tests.aux_tests
 	./run_bash_tests.sh
diff --git a/recipes/ljspeech/xtts_v1/train_gpt_xtts.py b/recipes/ljspeech/xtts_v1/train_gpt_xtts.py
index c170b1c828..94f3975c2f 100644
--- a/recipes/ljspeech/xtts_v1/train_gpt_xtts.py
+++ b/recipes/ljspeech/xtts_v1/train_gpt_xtts.py
@@ -93,6 +93,7 @@ def main():
         gpt_num_audio_tokens=8194,
         gpt_start_audio_token=8192,
         gpt_stop_audio_token=8193,
+        use_ne_hifigan=True, # if it is true it will keep the non-enhanced keys on the output checkpoint
     )
     # define audio config
     audio_config = XttsAudioConfig(
diff --git a/tests/tts_tests2/test_xtts_gpt_train.py b/tests/xtts_tests/test_xtts_gpt_train.py
similarity index 100%
rename from tests/tts_tests2/test_xtts_gpt_train.py
rename to tests/xtts_tests/test_xtts_gpt_train.py