From a24d7e663f0c3df2d08868aafcd4ab55af1ec321 Mon Sep 17 00:00:00 2001 From: lijiakun Date: Fri, 23 May 2025 15:39:18 +0800 Subject: [PATCH] fix refactor interfaces of mindformers --- .gitmodules | 2 +- .../models/mf_models/deepseekv3_weight_processor.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.gitmodules b/.gitmodules index d057201a..ec95f133 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 0861e60f..13b428ff 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 -- Gitee