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-12-17 23:17:12 -06:00
parent 9090c34f10
commit 9f2bd7f6e4
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
scripts/process_seed-tts.py
View File

@ -39,15 +39,15 @@ def process(
for line in lines:
if not line:
continue
speaker, text, prompt_wav, prompt_transcription = line.split("|")
filename, prompt_text, prompt_wav, text = line.split("|")
(output_dir / speaker).mkdir(parents=True, exist_ok=True)
(output_dir / filename / "out").mkdir(parents=True, exist_ok=True)
open( output_dir / speaker / "prompt.txt", "w", encoding="utf-8" ).write( text )
open( output_dir / speaker / "language.txt", "w", encoding="utf-8" ).write( language )
open( output_dir / filename / "prompt.txt", "w", encoding="utf-8" ).write( text )
open( output_dir / filename / "language.txt", "w", encoding="utf-8" ).write( language )
shutil.copy((input_dir / wav_dir / speaker).with_suffix(".wav"), output_dir / speaker / "reference.wav" )
shutil.copy(input_dir / prompt_wav, output_dir / speaker / "prompt.wav" )
shutil.copy((input_dir / wav_dir / filename).with_suffix(".wav"), output_dir / filename / "reference.wav" )
shutil.copy(input_dir / prompt_wav, output_dir / filename / "prompt.wav" )
if __name__ == "__main__":
process()