From a36142aa71a80ced84906c0b91224f8c6faf5941 Mon Sep 17 00:00:00 2001 From: b8zhong Date: Thu, 18 Dec 2025 20:15:33 -0800 Subject: [PATCH] Fix Mindspore model import warning (#15287) Co-authored-by: Even Zhou --- python/sglang/srt/models/mindspore.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/python/sglang/srt/models/mindspore.py b/python/sglang/srt/models/mindspore.py index 893364134..7d52872de 100644 --- a/python/sglang/srt/models/mindspore.py +++ b/python/sglang/srt/models/mindspore.py @@ -1,5 +1,7 @@ # SPDX-License-Identifier: Apache-2.0 # SPDX-FileCopyrightText: Copyright contributors to the SGLang project +from __future__ import annotations + import logging from typing import Any, Iterable, Optional, Tuple @@ -133,7 +135,7 @@ class LowerTriangularMask: def gen_attention_mask( self, is_prefill: bool, - position_ids: Tensor, + position_ids: "ms.Tensor", query_lens_np: np.ndarray, seq_lens_np: np.ndarray, ): @@ -289,7 +291,7 @@ class MindSporeForCausalLM(torch.nn.Module): input_ids: torch.Tensor, positions: torch.Tensor, forward_batch: ForwardBatch, - ) -> Tensor: + ) -> "ms.Tensor": # prepare base inputs model_inputs = self.prepare_inputs(input_ids, positions, forward_batch) # prepare model inputs