Skip to content

Commit

Permalink
Make style
Browse files Browse the repository at this point in the history
  • Loading branch information
erogol committed Sep 14, 2023
1 parent 44b61d2 commit ded7fd4
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 5 deletions.
4 changes: 2 additions & 2 deletions TTS/utils/manage.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
"apache 2.0": "https://choosealicense.com/licenses/apache-2.0/",
"apache2": "https://choosealicense.com/licenses/apache-2.0/",
"cc-by-sa 4.0": "https://creativecommons.org/licenses/by-sa/4.0/",
"cpml": "https://coqui.ai/cpml.txt"
"cpml": "https://coqui.ai/cpml.txt",
}


Expand Down Expand Up @@ -341,7 +341,7 @@ def download_model(self, model_name):
print(f" > {model_name} is already downloaded.")
else:
os.makedirs(output_path, exist_ok=True)
# handle TOS
# handle TOS
if not self.tos_agreed(model_item, output_path):
if not self.ask_tos(output_path):
os.rmdir(output_path)
Expand Down
9 changes: 6 additions & 3 deletions tests/zoo_tests/test_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@
from TTS.utils.generic_utils import get_user_data_dir
from TTS.utils.manage import ModelManager


MODELS_WITH_SEP_TESTS = ["bark", "xtts"]


Expand Down Expand Up @@ -70,7 +69,8 @@ def run_models(offset=0, step=1):
def test_xtts():
output_path = os.path.join(get_tests_output_path(), "output.wav")
speaker_wav = os.path.join(get_tests_data_path(), "ljspeech", "wavs", "LJ001-0001.wav")
run_cli("yes | "
run_cli(
"yes | "
f"tts --model_name tts_models/multilingual/multi-dataset/xtts_v1 "
f'--text "This is an example." --out_path "{output_path}" --progress_bar False --use_cuda True '
f'--speaker_wav "{speaker_wav}" --language_idx "en"'
Expand Down Expand Up @@ -98,9 +98,12 @@ def test_voice_conversion():
f" --out_path {output_path} --speaker_wav {speaker_wav} --reference_wav {reference_wav} --language_idx {language_id} --progress_bar False"
)


"""
These are used to split tests into different actions on Github.
"""


def test_models_offset_0_step_3():
run_models(offset=0, step=3)

Expand All @@ -110,4 +113,4 @@ def test_models_offset_1_step_3():


def test_models_offset_2_step_3():
run_models(offset=2, step=3)
run_models(offset=2, step=3)

0 comments on commit ded7fd4

Please sign in to comment.