xx

Author: UsernameFull

.github/workflows/ci-npu-mindspeed.yml

@@ -212,6 +212,6 @@ jobs:
       - name: Run MindSpeed import and collection smoke
         run: |
           export PYTHONPATH="${GITHUB_WORKSPACE}:${PYTHONPATH:-}"
-          python3 -m pytest tests/third_party/megatron --collect-only -q
+          python3 -m pytest tests/third_party/megatron
         env:
           ROLL_NPU_CI: "1"

tests/third_party/megatron/test_offload_states.py

@@ -36,7 +36,7 @@
 
 class McaModelCreator:
 
-    def __init__(self, optimizer_type, model_name="/data/cpfs_0/common/models/Qwen2.5-0.5B-Instruct"):
+    def __init__(self, optimizer_type, model_name="Qwen/Qwen2.5-0.5B-Instruct"):
         self.model_name = model_name
         if optimizer_type is None:
             self.megatron_train_args = TrainingArguments(