minor rollback gpt model change

Signed-off-by: jasonwan <[email protected]>
NVIDIA · Aug 31, 2023 · 8871d6c · 8871d6c
1 parent e43ce56
commit 8871d6c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/collections/nlp/models/language_modeling/megatron_gpt_model.py b/nemo/collections/nlp/models/language_modeling/megatron_gpt_model.py
@@ -1311,7 +1311,7 @@ def on_load_checkpoint(self, checkpoint) -> None:
                     key.replace('model.', ''): checkpoint_state_dict.pop(key)
                     for key in list(checkpoint_state_dict.keys())
                 }
-                missing_keys, unexpected_keys = module.load_state_dict(checkpoint_state_dict, strict=True)
+                module.load_state_dict(checkpoint_state_dict, strict=True)
 
         # legacy checkpointing for interleaved
         else: