UnboundLocalError: local variable 'gpt_cfg' referenced before assignment
https://github.com/NVIDIA/NeMo/blob/433d9f1d50ab11cea17ee8a5fd60c3911cc4ac47/examples/nlp/language_modeling/megatron_gpt_continue_training.py#L185
