Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix unit tests issue

This commit is contained in:
Edresson Casanova 2022-02-19 12:15:03 +00:00
parent fc7081fc5e
commit 5218d6b7a4
2 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/tts_tests/test_fast_pitch_speaker_emb_train.py
View File

@ -7,7 +7,7 @@ from TTS.config.shared_configs import BaseAudioConfig
from TTS.tts.configs.fast_pitch_config import FastPitchConfig from TTS.tts.configs.fast_pitch_config import FastPitchConfig
from TTS.utils.trainer_utils import get_last_checkpoint from TTS.utils.trainer_utils import get_last_checkpoint
config_path = os.path.join(get_tests_output_path(), "test_fast_pitch_config.json") config_path = os.path.join(get_tests_output_path(), "test_model_config.json")
output_path = os.path.join(get_tests_output_path(), "train_outputs") output_path = os.path.join(get_tests_output_path(), "train_outputs")
audio_config = BaseAudioConfig( audio_config = BaseAudioConfig(
@ -45,6 +45,8 @@ config = FastPitchConfig(
], ],
) )
config.audio.do_trim_silence = True config.audio.do_trim_silence = True
config.use_speaker_embedding = True
config.model_args.use_speaker_embedding = True
config.audio.trim_db = 60 config.audio.trim_db = 60
config.save_json(config_path) config.save_json(config_path)

6
tests/tts_tests/test_fast_pitch_train.py
View File

@ -7,7 +7,7 @@ from TTS.config.shared_configs import BaseAudioConfig
from TTS.tts.configs.fast_pitch_config import FastPitchConfig from TTS.tts.configs.fast_pitch_config import FastPitchConfig
from TTS.utils.trainer_utils import get_last_checkpoint from TTS.utils.trainer_utils import get_last_checkpoint
config_path = os.path.join(get_tests_output_path(), "test_fast_pitch_config.json") config_path = os.path.join(get_tests_output_path(), "test_model_config.json")
output_path = os.path.join(get_tests_output_path(), "train_outputs") output_path = os.path.join(get_tests_output_path(), "train_outputs")
audio_config = BaseAudioConfig( audio_config = BaseAudioConfig(
@ -42,8 +42,11 @@ config = FastPitchConfig(
test_sentences=[ test_sentences=[
"Be a voice, not an echo.", "Be a voice, not an echo.",
], ],
use_speaker_embedding=False,
) )
config.audio.do_trim_silence = True config.audio.do_trim_silence = True
config.use_speaker_embedding = False
config.model_args.use_speaker_embedding = False
config.audio.trim_db = 60 config.audio.trim_db = 60
config.save_json(config_path) config.save_json(config_path)
@ -58,6 +61,7 @@ command_train = (
"--coqpit.datasets.0.meta_file_attn_mask tests/data/ljspeech/metadata_attn_mask.txt " "--coqpit.datasets.0.meta_file_attn_mask tests/data/ljspeech/metadata_attn_mask.txt "
"--coqpit.test_delay_epochs 0" "--coqpit.test_delay_epochs 0"
) )
run_cli(command_train) run_cli(command_train)
# Find latest folder # Find latest folder