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

print warning if audio promtpless inferencing with low AR temp (it really doesn't like low temps / greedy sampling)

Browse Source
This commit is contained in:
mrq 2024-10-18 17:01:40 -05:00
parent c8f31db1de
commit 6b04c13c56
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vall_e/__main__.py
View File

@ -22,7 +22,7 @@ def main():
parser.add_argument("--max-nar-levels", type=int, default=7)
parser.add_argument("--ar-temp", type=float, default=0.0)
parser.add_argument("--nar-temp", type=float, default=0.01)
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)

3
vall_e/inference.py
View File

@ -270,6 +270,9 @@ class TTS():
return text_list[0]
# validate settings here
if not references and ar_temp < 0.5:
_logger.warning(f'Audio-promptless inferencing fails with low AR temperatures.')
for line in lines:
if out_path is None: