feat(cli+titan): support split known/unknown overrides and dynamic model patch

Author: Xiaoming-AMD

examples/megatron/prepare.py

@@ -191,7 +191,7 @@ def main():
 
     log_info(f"BACKEND_PATH {args.backend_path}")
     # primus_config = PrimusParser().parse(args)
-    primus_config = load_primus_config(args, unknown)
+    primus_config, _ = load_primus_config(args, unknown)
 
     primus_path = Path(args.primus_path).resolve()
     log_info(f"PRIMUS_PATH is set to: {primus_path}")

primus/modules/trainer/megatron/pre_trainer.py

@@ -120,10 +120,10 @@ def __init__(self, *args, **kwargs):
         kwargs["module_name"] = "pre_trainer"
 
         # Explicitly reject unknown extra_args
-        extra_args = kwargs.pop("extra_overrides", None)
+        extra_args = kwargs.pop("extra_args", None)
         if extra_args:
             raise ValueError(
-                f"[MegatronPretrainTrainer] Unexpected extra_overrides detected: {extra_args}. "
+                f"[MegatronPretrainTrainer] Unexpected extra_args detected: {extra_args}. "
                 f"Megatron backend does not support unregistered config keys."
             )