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

more demo page tweaks

Browse Source
This commit is contained in:
mrq 2024-07-21 19:31:13 -05:00
parent 7366f36f81
commit 3e5ca3a201
2 changed files with 35 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
data/demo/index.template.html
View File

@ -6,22 +6,31 @@
<h1>VALL-E Demo</h1>
<p>${PREAMBLE}</p>
<table>
<tr>
<th>Text</th>
<th>Prompt</th>
<th>Ground Truth</th>
<th>Our VALL-E</th>
<th>Original VALL-E</th>
<th>YourTTS</th>
</tr>${LIBRISPEECH_SAMPLES}
<thead>
<caption>LibriSpeech</caption>
<tr>
<th>Text</th>
<th>Prompt</th>
<th>Ground Truth</th>
<th>Our VALL-E</th>
<th>Original VALL-E</th>
<th>YourTTS</th>
</tr>
</thead>
<tbody>${LIBRISPEECH_SAMPLES}</tbody>
</table>
<table>
<tr>
<th>Text</th>
<th>Prompt</th>
<th>Ground Truth</th>
<th>Our VALL-E</th>
</tr>${DATASET_SAMPLES}
<thead>
<caption>Sampled Dataset</caption>
<tr>
<th>Text</th>
<th>Prompt</th>
<th>Ground Truth</th>
<th>Our VALL-E</th>
</tr>
</thead>
<tbody>${DATASET_SAMPLES}</tbody>
</table>
<p>Settings used: <pre>${SETTINGS}</pre></p>
</body>
</html>

13
vall_e/demo.py
View File

@ -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 = {