diff --git a/data/demo/index.template.html b/data/demo/index.template.html index 0c0224e..0773a94 100644 --- a/data/demo/index.template.html +++ b/data/demo/index.template.html @@ -6,22 +6,31 @@

VALL-E Demo

${PREAMBLE}

- - - - - - - - ${LIBRISPEECH_SAMPLES} + + + + + + + + + + + + ${LIBRISPEECH_SAMPLES}
TextPromptGround TruthOur VALL-EOriginal VALL-EYourTTS
LibriSpeech
TextPromptGround TruthOur VALL-EOriginal VALL-EYourTTS
- - - - - - ${DATASET_SAMPLES} + + + + + + + + + + ${DATASET_SAMPLES}
TextPromptGround TruthOur VALL-E
Sampled Dataset
TextPromptGround TruthOur VALL-E
+

Settings used:

${SETTINGS}

\ No newline at end of file diff --git a/vall_e/demo.py b/vall_e/demo.py index c13b019..46f201b 100644 --- a/vall_e/demo.py +++ b/vall_e/demo.py @@ -53,7 +53,7 @@ def main(): parser.add_argument("--nar-temp", type=float, default=0.0) parser.add_argument("--min-ar-temp", type=float, default=-1.0) parser.add_argument("--min-nar-temp", type=float, default=-1.0) - parser.add_argument("--input-prompt-length", type=float, default=3.0) + parser.add_argument("--input-prompt-length", type=float, default=0.0) parser.add_argument("--top-p", type=float, default=1.0) parser.add_argument("--top-k", type=int, default=16) @@ -89,6 +89,17 @@ def main(): # replace values in our template html = html.replace(r"${PREAMBLE}", args.preamble ) + html = html.replace(r"${SETTINGS}", str(dict( + 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, + min_ar_temp=args.min_ar_temp, min_nar_temp=args.min_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, + )) ) # pull from provided samples samples_dirs = {