From b06f6cd224489a9d69f3e5edf456c68cf7385704 Mon Sep 17 00:00:00 2001 From: HeyyyyyyG <49757268+HeyyyyyyG@users.noreply.github.com> Date: Wed, 7 Feb 2024 18:34:05 -0800 Subject: [PATCH] load_from_nemo bug fix (#103) Signed-off-by: Jiaqi Zeng --- nemo_aligner/utils/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nemo_aligner/utils/utils.py b/nemo_aligner/utils/utils.py index 74198d476..70d6820ab 100644 --- a/nemo_aligner/utils/utils.py +++ b/nemo_aligner/utils/utils.py @@ -97,7 +97,7 @@ def load_from_nemo( save_restore_connector=connector, strict=strict, ) - return model, model_cfg if return_updated_cfg else model + return (model, model_cfg) if return_updated_cfg else model def load_checkpoint_model_config(restore_path):