fake_init return value fix

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>
NVIDIA · Jun 3, 2024 · 4bf4928 · 4bf4928
1 parent 247db89
commit 4bf4928
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/nemo/collections/multimodal/parts/utils.py b/nemo/collections/multimodal/parts/utils.py
@@ -391,6 +391,7 @@ def create_neva_model_and_processor(cfg):
             app_state.tensor_model_parallel_size = cfg.tensor_model_parallel_size
             app_state.pipeline_model_parallel_size = cfg.pipeline_model_parallel_size
             (
+                _,
                 app_state.tensor_model_parallel_rank,
                 app_state.pipeline_model_parallel_rank,
                 app_state.expert_model_parallel_rank,