diff --git a/codecheck_toolkits/vllm_codecheck.sh b/codecheck_toolkits/vllm_codecheck.sh index c91087fe19df09b701de0b66d52a64f008ccb037..6555b030dc4374d6ca4a644f683e61f9f120ef15 100644 --- a/codecheck_toolkits/vllm_codecheck.sh +++ b/codecheck_toolkits/vllm_codecheck.sh @@ -60,6 +60,16 @@ fi PYTHON_VERSION=$(python -c 'import sys; print(f"{sys.version_info.major}.{sys.version_info.minor}")') +# download vllm + +cd codecheck_toolkits +git clone https://gitee.com/moran3/vllm.git -b v0.8.3 +cd - + +export MYPYPATH=codecheck_toolkits/vllm + +# check mypy + if ! git diff --cached --diff-filter=ACM --quiet --exit-code "$MERGEBASE" -- '*.py' '*.pyi' &> /dev/null; then git diff --cached --name-only --diff-filter=ACM "$MERGEBASE" -- '*.py' '*.pyi' | xargs \ mypy --follow-imports skip --python-version "${PYTHON_VERSION}" "$@" 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 2ee13c23816a3462331e19526a8904c580a86ec2..1f01d8dff037442328ea4cecaa8c720100208fad 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 @@ -1597,7 +1597,10 @@ class DeepseekV3WeightProcessor(BaseWeightProcessor): ('quant' not in file and (not self.is_quant or is_mtp_model)): param_json_path = os.path.join(src_hf_dir, file) with open(param_json_path, "r") as fp: - hf_weight_map = json.load(fp)['weight_map'] + hf_weight_map = json.load(fp) + if not hf_weight_map.get('weight_map'): + continue + hf_weight_map = hf_weight_map['weight_map'] break elif file.endswith('_name_map.json'): param_json_path = os.path.join(src_hf_dir, file)