From fd91e447a151dd2f5b9dd8d0f1cb57c7fb9acb7b Mon Sep 17 00:00:00 2001 From: mrq Date: Wed, 26 Feb 2025 10:46:02 -0600 Subject: [PATCH] lol --- vall_e/engines/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vall_e/engines/__init__.py b/vall_e/engines/__init__.py index 955f50a..2a28f1d 100755 --- a/vall_e/engines/__init__.py +++ b/vall_e/engines/__init__.py @@ -146,7 +146,7 @@ def load_engines(training=True, **model_kwargs): elif cfg.hyperparameters.optimizer.lower() == "adagrad": optimizer_class = ml.Adagrad elif cfg.hyperparameters.optimizer.lower() == "muon": - optimizer = ml.Muon + optimizer_class = ml.Muon muon_params = [ param for name, param in model.model.named_parameters() if param.ndim >= 2 ] adamw_params = [ param for name, param in model.model.named_parameters() if param.ndim < 2 ]