diff --git a/.gitmodules b/.gitmodules
index d057201a76ae4ad81c3deff14ede37dbad233438..ec95f133e4e555afa289119a89f145fa7f17942f 100644
--- a/.gitmodules
+++ b/.gitmodules
@@ -1,4 +1,4 @@
 [submodule "tests/mindformers"]
 	path = tests/mindformers
 	url = https://gitee.com/mindspore/mindformers.git
-	branch = br_infer_deepseek_os
+	branch = dev
diff --git a/vllm_mindspore/model_executor/models/mf_models/deepseekv3_weight_processor.py b/vllm_mindspore/model_executor/models/mf_models/deepseekv3_weight_processor.py
index 0861e60f73c0775338e106d42de0acb1c56e1461..13b428ffa5e09a41b6aaa4e23d44ee24d3fcf0c4 100644
--- a/vllm_mindspore/model_executor/models/mf_models/deepseekv3_weight_processor.py
+++ b/vllm_mindspore/model_executor/models/mf_models/deepseekv3_weight_processor.py
@@ -25,7 +25,7 @@ from tqdm import tqdm
 import mindspore as ms
 from mindspore import dtype
 from mindspore.communication.management import get_rank
-from mindformers.experimental.parallel_core.pynative.parallel_state import get_tensor_model_parallel_rank
+from mindformers.parallel_core.inference.parallel_state import get_tensor_model_parallel_rank
 from vllm_mindspore.model_executor.models.mf_models.weight_processor import BaseWeightProcessor
 from vllm_mindspore.utils import convert_np_to_ms_dtype