From 02dfc60ac35a1d4a81e1c66f6bf1cd68c7b912ce Mon Sep 17 00:00:00 2001 From: mrq Date: Fri, 18 Oct 2024 17:23:22 -0500 Subject: [PATCH] ugh --- vall_e/demo.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/vall_e/demo.py b/vall_e/demo.py index 1329af1..2003ffe 100644 --- a/vall_e/demo.py +++ b/vall_e/demo.py @@ -266,7 +266,7 @@ def main(): audio_samples = [ prompt, out_path ] if args.comparison: audio_samples += [ out_path_comparison ] - audio_samples += [ p for p in external_sources if p.exists() else None ] + audio_samples += [ p if p.exists() else None for p in external_sources ] if not args.random_prompts or k == "librispeech": audio_samples += [ reference ] @@ -309,15 +309,18 @@ def main(): return # swap model config swap + """ if "dtype" in kwargs or "amp" in kwargs: dtype = kwargs.pop("dtype", args.dtype) amp = kwargs.pop("amp", args.amp) del tts tts = TTS( config=args.yaml, device=args.device, dtype=dtype, amp=amp ) + """ try: tts.inference( out_path=out_path, **kwargs ) except Exception as e: + raise e print(f'Error while processing {out_path}: {e}') if args.comparison: