[BugFix][Qwen3-Omni]Fixed the issue of incorrect answers for single words. (#2239)

Signed-off-by: amy-why-3459 <wuhaiyan17@huawei.com>
This commit is contained in:
amy-why-3459 authored and GitHub committed 2026-03-27 11:46:24 +08:00
1 parent 6f87a9382f
commit 9f667ee151
2 files changed
+2 -5

No files matched your search

+1 -2
View File
@@ -376,9 +376,8 @@ async def benchmark(
limit_per_host=max_concurrency or 0,
ttl_dns_cache=300,
use_dns_cache=True,
keepalive_timeout=60,
enable_cleanup_closed=True,
force_close=False,
force_close=True,
ssl=ssl_setting,
)
@@ -1047,9 +1047,7 @@ class Qwen3OmniMoeForConditionalGeneration(
dim=0,
)
else:
trailing_text_hidden = torch.zeros(
tts_eos_embed.shape, device=tts_eos_embed.device, dtype=tts_eos_embed.dtype
)
trailing_text_hidden = tts_eos_embed
input_embeds = assistant_text_hidden + assistant_codec_hidden
input_ids = torch.full(