mrq/vall-e
5
9
Fork 0
Code Issues 8 Pull Requests Packages Projects Releases Wiki Activity

ugh

Browse Source
This commit is contained in:
mrq 2024-10-18 17:23:22 -05:00
parent 71731ed785
commit 02dfc60ac3
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
vall_e/demo.py
View File

@ -266,7 +266,7 @@ def main():
audio_samples = [ prompt, out_path ] audio_samples = [ prompt, out_path ]
if args.comparison: if args.comparison:
audio_samples += [ out_path_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": if not args.random_prompts or k == "librispeech":
audio_samples += [ reference ] audio_samples += [ reference ]
@ -309,15 +309,18 @@ def main():
return return
# swap model config swap # swap model config swap
"""
if "dtype" in kwargs or "amp" in kwargs: if "dtype" in kwargs or "amp" in kwargs:
dtype = kwargs.pop("dtype", args.dtype) dtype = kwargs.pop("dtype", args.dtype)
amp = kwargs.pop("amp", args.amp) amp = kwargs.pop("amp", args.amp)
del tts del tts
tts = TTS( config=args.yaml, device=args.device, dtype=dtype, amp=amp ) tts = TTS( config=args.yaml, device=args.device, dtype=dtype, amp=amp )
"""
try: try:
tts.inference( out_path=out_path, **kwargs ) tts.inference( out_path=out_path, **kwargs )
except Exception as e: except Exception as e:
raise e
print(f'Error while processing {out_path}: {e}') print(f'Error while processing {out_path}: {e}')
if args.comparison: if args.comparison: