Small cleanup

Signed-off-by: Rafael Vasquez <rafvasq21@gmail.com>

Author: rafvasq

aiu_fms_testing_utils/scripts/drive_paged_programs.py

@@ -362,7 +362,6 @@ def __load_validation_info(
         dist.get_rank() == 0
     )
 
-# not validation
 with stagger_region(args.stagger_load):
     model = get_model(
         architecture="hf_pretrained",
@@ -379,7 +378,6 @@ def __load_validation_info(
 
 __maybe_prepare_fp8_weights(model, is_fp8)
 
-# is validation
 if not args.skip_validation:
     with stagger_region(args.stagger_load):
         validation_model = get_model(

aiu_fms_testing_utils/scripts/refactored_dpp.py

@@ -941,4 +941,4 @@ def process_tokens_test(
                 attn_name=ATTN_NAME, 
                 cpu_dtype=CPU_DTYPE, 
                 tokenizer=tokenizer,
-            )
+            )