[Hardware][Ascend][Model] Support MOSS-TTS-Nano on Ascend A2 (#7192)

Signed-off-by: Big2Wheel <1404207590@qq.com>
This commit is contained in:
yulun authored and GitHub committed 2026-09-14 15:25:17 -07:00
1 parent c68eadbbe0
commit 8e088c7725
8 files changed
+224 -22

No files matched your search

@@ -0,0 +1,32 @@
# SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
import pytest
import torch
from vllm_omni.model_executor.models.moss_tts_nano.modeling_moss_tts_nano import (
_configure_attention_implementation,
_validate_max_num_seqs,
)
pytestmark = [pytest.mark.core_model, pytest.mark.cpu]
def test_non_cuda_attention_uses_sdpa():
class FakeModel:
attention_implementation = None
def _set_attention_implementation(self, implementation):
self.attention_implementation = implementation
model = FakeModel()
_configure_attention_implementation(model, torch.device("cpu"))
assert model.attention_implementation == "sdpa"
def test_moss_tts_nano_rejects_concurrent_active_sequences():
_validate_max_num_seqs(1)
with pytest.raises(ValueError, match="requires max_num_seqs=1"):
_validate_max_num_seqs(2)