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

cleanup the resultant text from STT

Browse Source
This commit is contained in:
mrq 2024-09-06 18:44:25 -05:00
parent fa93061b3e
commit a6ad0577b8
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vall_e/inference.py
View File

@ -210,7 +210,7 @@ class TTS():
else:
raise Exception("!")
text_list = [ cfg.tokenizer.decode( text ) for text in text_list ]
text_list = [ cfg.tokenizer.decode( text ).replace(" ", "_").replace(" ", "").replace("_", " ") for text in text_list ]
print( text_list )
return text_list[0]

2
vall_e/webui.py
View File

@ -124,7 +124,7 @@ def do_inference_tts( progress=gr.Progress(track_tqdm=True), *args, **kwargs ):
parser = argparse.ArgumentParser(allow_abbrev=False)
# I'm very sure I can procedurally generate this list
parser.add_argument("--text", type=str, default=kwargs["text"])
parser.add_argument("--task", type=str, default=kwargs["task"])
parser.add_argument("--task", type=str, default="tts")
parser.add_argument("--references", type=str, default=kwargs["reference"])
parser.add_argument("--language", type=str, default="en")
parser.add_argument("--input-prompt-length", type=float, default=kwargs["input-prompt-length"])