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 2025-02-24 13:54:21 -06:00
parent 8f5a3997bd
commit 6e7b269147
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
vall_e/train.py
View File

@ -207,10 +207,10 @@ def run_eval(engines, eval_name, dl, args=None):
training=False, training=False,
) )
if self.version >= 7: if engine.hyper_config.version >= 7:
kwargs = base_kwargs | cfg.evaluation.kwargs kwargs = base_kwargs | cfg.evaluation.kwargs
# sample for NAR demask # sample for NAR demask
if random.random() < cfg.model.experimental.masking_train_p: if random.random() < engine.hyper_config.experimental.masking_train_p:
kwargs["len_list"] = [ resp.shape[0] for resp in batch["resps"] ] kwargs["len_list"] = [ resp.shape[0] for resp in batch["resps"] ]
# inference # inference
resps_list = engine( **kwargs ) resps_list = engine( **kwargs )