From 8c04813bc74cf3df199a22f3975038fe589ad0a0 Mon Sep 17 00:00:00 2001 From: zhang_xu_hao1230 Date: Mon, 14 Jul 2025 20:11:45 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8Dmss=20=E5=BC=82=E5=B8=B8?= =?UTF-8?q?=E5=9C=BA=E6=99=AF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- vllm_mindspore/model_executor/models/model_base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm_mindspore/model_executor/models/model_base.py b/vllm_mindspore/model_executor/models/model_base.py index 9df5447e..bb2213cc 100644 --- a/vllm_mindspore/model_executor/models/model_base.py +++ b/vllm_mindspore/model_executor/models/model_base.py @@ -306,7 +306,7 @@ class MsModelBase: model_inputs["input_ids"] = input_ids.astype(ms.int32) * 1 model_inputs["block_tables"] = attn_metadata.block_tables * 1 model_inputs["batch_valid_length"] = ms.from_numpy(seq_lens_np) - model_inputs["slot_mapping"] = attn_metadata.slot_mapping + model_inputs["slot_mapping"] = attn_metadata.slot_mapping * 1 model_inputs["position_ids"] = position_ids model_inputs["q_seq_lens"] = q_seq_lens model_inputs["attention_mask"] = attention_mask -- Gitee