fix issues

Author: jiaqiw09

examples/ascend/megatron/train_sft_full.sh

@@ -21,7 +21,7 @@ megatron sft \
     --lr 1e-5 \
     --lr_warmup_fraction 0.05 \
     --min_lr 1e-6 \
-    --output_dir megatron_output/Qwen3-32B \
+    --output_dir megatron_output/Qwen3-4B \
     --eval_steps 500 \
     --save_steps 500 \
     --max_length 8192 \
@@ -30,4 +30,4 @@ megatron sft \
     --no_save_optim true \
     --no_save_rng true \
     --sequence_parallel true \
-    --attention_backend flash
+    --attention_backend flash

swift/megatron/model/model_config.py

@@ -189,10 +189,10 @@ def _augment_mindspeed_defaults(self):
         if MegatronModelConfig._mindspeed_defaults_cache is None:
             defaults = {}
             try:
-                from mindspeed.arguments import process_args
-                from argparse import ArgumentParser
                 import mindspeed.features_manager as mfm
                 import sys
+                from argparse import ArgumentParser
+                from mindspeed.arguments import process_args
 
                 original_features = list(mfm.FEATURES_LIST)
                 full_features = mfm.create_features_list()
@@ -202,20 +202,21 @@ def _augment_mindspeed_defaults(self):
                     parser = ArgumentParser()
                     process_args(parser)
                     # Parse args from sys.argv
-                    args, _ = parser.parse_known_args()
+                    args, _ = parser.parse_known_args([])
                     defaults = vars(args)
                 finally:
                     mfm.FEATURES_LIST.clear()
                     mfm.FEATURES_LIST.extend(original_features)
-            except Exception:
+            except Exception as e:
+                logger.warning(f'Failed to get MindSpeed defaults, which may cause issues on NPU: {e}')
                 defaults = {}
             MegatronModelConfig._mindspeed_defaults_cache = defaults
 
         for name, value in MegatronModelConfig._mindspeed_defaults_cache.items():
             if not hasattr(self, name):
                 setattr(self, name, value)
             elif hasattr(self, name) and getattr(self, name) is None and value is not None:
-                 setattr(self, name, value)
+                setattr(self, name, value)
 
     def __post_init__(self):
         self._augment_mindspeed_defaults()