[train]: add apply_dsa_fusion and original_max_position_embeddings arguments.

Author: LiJunscs

flagscale/train/megatron/training/arguments.py

@@ -2073,6 +2073,7 @@ def _add_network_size_args(parser):
         "persist_layer_norm",
         "bias_dropout_fusion",
         "apply_rope_fusion",
+        "apply_dsa_kernel_fusion",
         # FlagScale-specific: manually added in arguments_fs.py
         "enable_hetero",
         "hetero_pipeline_layer_split",
@@ -3217,6 +3218,12 @@ def _add_mla_args(parser):
                        help="Dimension of the head in the V projection.")
     group.add_argument('--rotary-scaling-factor', type=float, default=1.0,
                        help="Rotary scaling factor for the rotary embeddings.")
+    group.add_argument(
+        '--original-max-position-embeddings',
+        type=int,
+        default=4096,
+        help="Original maximum position embeddings for the original model, used by yarn.",
+    )
     group.add_argument('--mscale', type=float, default=1.0,
                        help="Mscale for YaRN RoPE in multi-latent attention.")
     group.add_argument('--mscale-all-dim', type=float, default=0.0,