From 6e7b26914719698ab126b5c6c01a09f896199081 Mon Sep 17 00:00:00 2001 From: mrq Date: Mon, 24 Feb 2025 13:54:21 -0600 Subject: [PATCH] ugh --- vall_e/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vall_e/train.py b/vall_e/train.py index 0257c2a..4c2943d 100755 --- a/vall_e/train.py +++ b/vall_e/train.py @@ -207,10 +207,10 @@ def run_eval(engines, eval_name, dl, args=None): training=False, ) - if self.version >= 7: + if engine.hyper_config.version >= 7: kwargs = base_kwargs | cfg.evaluation.kwargs # 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"] ] # inference resps_list = engine( **kwargs )