auracaster/piper
3
0
Fork 0
mirror of https://github.com/pstrueb/piper.git synced 2026-04-21 15:24:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove executor

Browse Source
This commit is contained in:
Michael Hansen
2023-04-09 11:38:11 -05:00
parent d88965181b
commit 3d5d53f07c
1 changed files with 22 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
src/python/piper_train/preprocess.py
View File

@@ -192,36 +192,30 @@ def process_batch(args: argparse.Namespace, queue_in: JoinableQueue, queue_out:
silence_detector = make_silence_detector()
phonemizer = Phonemizer(default_voice=args.language)
with ThreadPoolExecutor(max_workers=1) as executor:
while True:
utt_batch = queue_in.get()
if utt_batch is None:
break
while True:
utt_batch = queue_in.get()
if utt_batch is None:
break
for utt in utt_batch:
try:
utt.phonemes = next(
executor.map(
lambda utt: phonemize(utt.text, phonemizer),
[utt],
timeout=1,
)
)
utt.phoneme_ids = phonemes_to_ids(utt.phonemes)
utt.audio_norm_path, utt.audio_spec_path = cache_norm_audio(
utt.audio_path,
args.cache_dir,
silence_detector,
args.sample_rate,
)
queue_out.put(utt)
except TimeoutError:
_LOGGER.error("Skipping utterance due to timeout: %s", utt)
except Exception:
_LOGGER.exception("Failed to process utterance: %s", utt)
queue_out.put(None)
for utt in utt_batch:
try:
_LOGGER.debug(utt)
utt.phonemes = phonemize(utt.text, phonemizer)
utt.phoneme_ids = phonemes_to_ids(utt.phonemes)
utt.audio_norm_path, utt.audio_spec_path = cache_norm_audio(
utt.audio_path,
args.cache_dir,
silence_detector,
args.sample_rate,
)
queue_out.put(utt)
except TimeoutError:
_LOGGER.error("Skipping utterance due to timeout: %s", utt)
except Exception:
_LOGGER.exception("Failed to process utterance: %s", utt)
queue_out.put(None)
queue_in.task_done()
queue_in.task_done()
except Exception:
_LOGGER.exception("process_batch")