57 lines
2.2 KiB
Python
Executable File
57 lines
2.2 KiB
Python
Executable File
import argparse
|
|
from pathlib import Path
|
|
from .inference import TTS
|
|
|
|
def path_list(arg):
|
|
return [Path(p) for p in arg.split(";")]
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser("VALL-E TTS")
|
|
parser.add_argument("text")
|
|
parser.add_argument("references", type=path_list)
|
|
parser.add_argument("--out-path", type=Path, default=None)
|
|
|
|
parser.add_argument("--yaml", type=Path, default=None)
|
|
parser.add_argument("--ar-ckpt", type=Path, default=None)
|
|
parser.add_argument("--nar-ckpt", type=Path, default=None)
|
|
|
|
parser.add_argument("--max-ar-steps", type=int, default=6 * 75)
|
|
parser.add_argument("--max-nar-levels", type=int, default=7)
|
|
|
|
parser.add_argument("--ar-temp", type=float, default=1.0)
|
|
parser.add_argument("--nar-temp", type=float, default=1.0)
|
|
parser.add_argument("--input-prompt-length", type=float, default=3.0)
|
|
|
|
parser.add_argument("--top-p", type=float, default=1.0)
|
|
parser.add_argument("--top-k", type=int, default=0)
|
|
parser.add_argument("--repetition-penalty", type=float, default=1.0)
|
|
parser.add_argument("--repetition-penalty-decay", type=float, default=0.0)
|
|
parser.add_argument("--length-penalty", type=float, default=0.0)
|
|
parser.add_argument("--beam-width", type=int, default=0)
|
|
|
|
parser.add_argument("--mirostat-tau", type=float, default=0)
|
|
parser.add_argument("--mirostat-eta", type=float, default=0)
|
|
|
|
parser.add_argument("--device", type=str, default=None)
|
|
parser.add_argument("--amp", action="store_true")
|
|
parser.add_argument("--dtype", type=str, default=None)
|
|
args = parser.parse_args()
|
|
|
|
tts = TTS( config=args.yaml, ar_ckpt=args.ar_ckpt, nar_ckpt=args.nar_ckpt, device=args.device, dtype=args.dtype, amp=args.amp )
|
|
tts.inference(
|
|
text=args.text,
|
|
references=args.references,
|
|
out_path=args.out_path,
|
|
input_prompt_length=args.input_prompt_length,
|
|
max_ar_steps=args.max_ar_steps, max_nar_levels=args.max_nar_levels,
|
|
ar_temp=args.ar_temp, nar_temp=args.nar_temp,
|
|
top_p=args.top_p, top_k=args.top_k,
|
|
repetition_penalty=args.repetition_penalty, repetition_penalty_decay=args.repetition_penalty_decay,
|
|
length_penalty=args.length_penalty,
|
|
beam_width=args.beam_width,
|
|
mirostat_tau=args.mirostat_tau, mirostat_eta=args.mirostat_eta
|
|
)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|