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

ugh

Browse Source
This commit is contained in:
mrq 2024-06-08 16:14:57 -05:00
parent b072f9b96b
commit ead3e2f0cb
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vall_e/train.py
View File

@ -169,7 +169,7 @@ def run_eval(engines, eval_name, dl):
resps_list[i] = torch.stack( resp ).t()
else:
if "len" in engine.hyper_config.capabilities:
len_list = engine(text_list=batch["text"], proms_list=batch["proms"], max_steps=steps )
len_list = engine(text_list=batch["text"], proms_list=batch["proms"], max_steps=10 ) # don't need more than that
resps_list = engine( text_list=batch["text"], proms_list=batch["proms"], len_list=len_list )
else:
if "ar" in engine.hyper_config.capabilities: