fixes

Signed-off-by: adil-a <adil.asif2000@hotmail.com>

Author: adil-a

examples/llm_finetune/nemotron/nemotron_nano_9b_squad.yaml

@@ -38,6 +38,7 @@ model:
   _target_: nemo_automodel.NeMoAutoModelForCausalLM.from_pretrained
   pretrained_model_name_or_path: nvidia/NVIDIA-Nemotron-Nano-9B-v2
   trust_remote_code: true
+  force_hf: true
 
 # torch.compile configuration
 compile:

examples/llm_finetune/nemotron/nemotron_nano_9b_squad_peft.yaml

@@ -38,6 +38,7 @@ model:
   _target_: nemo_automodel.NeMoAutoModelForCausalLM.from_pretrained
   pretrained_model_name_or_path: nvidia/NVIDIA-Nemotron-Nano-9B-v2
   trust_remote_code: true
+  force_hf: true
 
 # torch.compile configuration
 compile:

examples/llm_finetune/nemotron/nemotron_nano_v3_hellaswag.yaml

@@ -14,9 +14,9 @@
 
 
 # To run this recipe, please use the following command. Replace --nproc-per-node=8 with the number of GPUs on your node:
-# torchrun --nproc-per-node=8 examples/llm_finetune/finetune.py --config examples/llm_finetune/nemotron/nemotron_nano_v3_squad.yaml
+# torchrun --nproc-per-node=8 examples/llm_finetune/finetune.py --config examples/llm_finetune/nemotron/nemotron_nano_v3_hellaswag.yaml
 # For multi-node, use the following:
-# torchrun --nproc-per-node=8 --nnodes=2 examples/llm_finetune/finetune.py --config examples/llm_finetune/nemotron/nemotron_nano_v3_squad.yaml
+# torchrun --nproc-per-node=8 --nnodes=2 examples/llm_finetune/finetune.py --config examples/llm_finetune/nemotron/nemotron_nano_v3_hellaswag.yaml
 
 
 step_scheduler:

examples/llm_finetune/nemotron/nemotron_nano_v3_hellaswag_peft.yaml

@@ -13,10 +13,8 @@
 # limitations under the License.
 
 
-# To run this recipe, please use the following command. Replace --nproc-per-node=8 with the number of GPUs on your node:
-# torchrun --nproc-per-node=8 examples/llm_finetune/finetune.py --config examples/llm_finetune/nemotron/nemotron_nano_v3_squad_peft.yaml
-# For multi-node, use the following:
-# torchrun --nproc-per-node=8 --nnodes=2 examples/llm_finetune/finetune.py --config examples/llm_finetune/nemotron/nemotron_nano_v3_squad_peft.yaml
+# To run this recipe, please use the following command:
+# python examples/llm_finetune/finetune.py --config examples/llm_finetune/nemotron/nemotron_nano_v3_hellaswag_peft.yaml
 
 
 step_scheduler:
@@ -39,6 +37,7 @@ model:
   _target_: nemo_automodel.NeMoAutoModelForCausalLM.from_pretrained
   pretrained_model_name_or_path: nvidia/NVIDIA-Nemotron-3-Nano-30B-A3B-BF16
   trust_remote_code: true
+  force_hf: true
 
 # torch.compile configuration
 compile: