AttributeError: type object 'MegatronGPTSFTModel' has no attribute 'merge_cfg_with'

[ssyrc]

Describe the bug

Error while pretraining Llama3.1-8B-Nemotron-Nano-v1 in .nemo format with custom dataset

• Error code:
  python /opt/NeMo/examples/nlp/language_modeling/megatron_gpt_pretraining.py $configs

• Error Message:
  Traceback (most recent call last): File "/opt/NeMo/examples/nlp/language_modeling/megatron_gpt_pretraining.py", line 49, in main model_cfg = MegatronGPTSFTModel.merge_cfg_with(cfg.model.restore_from_path, cfg) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ AttributeError: type object 'MegatronGPTSFTModel' has no attribute 'merge_cfg_with'

Environment details

• Nemo container ver: nvcr.io/nvidia/nemo:25.07.nemotron-nano-v2

[cuichenx]

Hi @ssyrc, please check the documentation here for training with NeMo 2. https://docs.nvidia.com/nemo-framework/user-guide/latest/llms/llama_nemotron.html

Scripts in the nlp collections have been deprecated since 24.09