jyapayne/mimic3
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use inference settings from config

Browse source
This commit is contained in:
Michael Hansen 2022-03-21 16:52:32 -04:00
commit 355a467c5e
2 changed files with 27 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

16
mimic3-tts/mimic3_tts/tts.py
View file

@ -134,6 +134,21 @@ class Mimic3TextToSpeechSystem(TextToSpeechSystem):
continue
voice_lang = lang_dir.name
# Load config
config_path = voice_dir / "config.json"
_LOGGER.debug("Loading config from %s", config_path)
with open(config_path, "r", encoding="utf-8") as config_file:
config = TrainingConfig.load(config_file)
properties: typing.Dict[str, typing.Any] = {
"length_scale": config.inference.length_scale,
"noise_scale": config.inference.noise_scale,
"noise_w": config.inference.noise_w,
}
# Load speaker names
voice_name = voice_dir.name
speakers: typing.Optional[typing.Sequence[str]] = None
@ -154,6 +169,7 @@ class Mimic3TextToSpeechSystem(TextToSpeechSystem):
language=voice_lang,
description="",
speakers=speakers,
properties=properties,
)
def begin_utterance(self):