missing imports, updated names

Author: mollys

src/megatron/bridge/perf_recipes/llama/__init__.py

@@ -46,6 +46,13 @@
     llama31_405b_pretrain_256gpu_b300_fp8mx_config,
     llama31_405b_pretrain_256gpu_b300_nvfp4_config,
 )
+from megatron.bridge.perf_recipes.llama.gb200.llama2 import (
+    llama2_70b_peft_4gpu_gb200_fp8ds_config,
+    llama2_70b_peft_8gpu_gb200_fp8ds_config,
+    llama2_70b_peft_8gpu_gb200_nvfp4_config,
+    llama2_70b_peft_72gpu_gb200_fp8ds_config,
+    llama2_70b_peft_512gpu_gb200_fp8ds_config,
+)
 from megatron.bridge.perf_recipes.llama.gb200.llama3 import (
     llama3_8b_pretrain_8gpu_gb200_bf16_config,
     llama3_8b_pretrain_8gpu_gb200_fp8cs_config,
@@ -79,6 +86,12 @@
     llama31_405b_pretrain_256gpu_gb200_fp8mx_config,
     llama31_405b_pretrain_256gpu_gb200_nvfp4_config,
 )
+from megatron.bridge.perf_recipes.llama.gb300.llama2 import (
+    llama2_70b_peft_4gpu_gb300_fp8ds_config,
+    llama2_70b_peft_8gpu_gb300_fp8ds_config,
+    llama2_70b_peft_72gpu_gb300_fp8ds_config,
+    llama2_70b_peft_512gpu_gb300_fp8ds_config,
+)
 from megatron.bridge.perf_recipes.llama.gb300.llama3 import (
     llama3_8b_pretrain_8gpu_gb300_bf16_config,
     llama3_8b_pretrain_8gpu_gb300_fp8cs_config,

src/megatron/bridge/perf_recipes/llama/common.py

@@ -19,6 +19,9 @@
     _enable_overlap_param_gather_with_optimizer_step,
     _perf_precision,
 )
+from megatron.bridge.recipes.llama.llama2 import (
+    llama2_70b_peft_config,
+)
 from megatron.bridge.recipes.llama.llama3 import (
     llama3_8b_pretrain_config,
     llama3_8b_sft_config,

src/megatron/bridge/recipes/llama/h100/llama2.py

@@ -17,7 +17,7 @@
 from megatron.bridge import AutoBridge
 from megatron.bridge.peft.base import PEFT
 from megatron.bridge.recipes.common import _peft_common, _pretrain_common
-from megatron.bridge.recipes.utils.finetune_utils import default_peft_confi
+from megatron.bridge.recipes.utils.dataset_utils import default_peft_config
 from megatron.bridge.recipes.utils.environment_utils import COMMON_RECIPE_ENV_VARS
 from megatron.bridge.recipes.utils.tokenizer_utils import DEFAULT_NULL_TOKENIZER_VOCAB_SIZE
 from megatron.bridge.training.comm_overlap import CommOverlapConfig
@@ -152,7 +152,7 @@ def llama2_70b_peft_config(
     seq_length = 8192
     cfg.model.seq_length = seq_length
     cfg.dataset.seq_length = seq_length
-    cfg.dataset.packed_sequence_specs.packed_sequence_size = seq_length
+    cfg.dataset.offline_packing_specs.packed_sequence_size = seq_length
 
     # PEFT config - 70B uses dim=16, alpha=32
     peft_cfg = default_peft_config(peft_scheme)
@@ -164,7 +164,7 @@ def llama2_70b_peft_config(
 
     # Packed sequence settings
     if cfg.model.context_parallel_size > 1:
-        cfg.dataset.packed_sequence_specs.pad_seq_to_mult = cfg.model.context_parallel_size * 2
+        cfg.dataset.offline_packing_specs.pad_seq_to_mult = cfg.model.context_parallel_size * 2
 
     # Parallelism settings - 70B PEFT uses TP=8
     cfg.model.pipeline_model_parallel_layout = None