load_from_nemo bug fix (#103)

Signed-off-by: Jiaqi Zeng <[email protected]>
NVIDIA · Feb 8, 2024 · b06f6cd · b06f6cd
1 parent 1e4f4cf
commit b06f6cd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):