Temp fix for metrics error in test_qwen3_tts

Signed-off-by: tzhouam <tzhouam@connect.ust.hk>
This commit is contained in:
tzhouam committed 2026-02-12 10:02:19 +00:00
1 parent db40563b88
commit f8959bcc6e
2 files changed
+5 -1

No files matched your search

+2 -1
View File
@@ -20,7 +20,7 @@ import pytest
from tests.conftest import OmniServer
from tests.utils import hardware_test
MODEL = "Qwen/Qwen3-TTS-12Hz-0.6B-CustomVoice"
MODEL = "/scratch/dyvm6xra/dyvm6xrauser49/rebase/models/hub/models--Qwen--Qwen3-TTS-12Hz-0.6B-CustomVoice/snapshots/85e237c12c027371202489a0ec509ded67b5e4b5"
def get_stage_config():
@@ -46,6 +46,7 @@ def omni_server():
"120",
"--trust-remote-code",
"--enforce-eager",
"--disable-log-stats",
],
) as server:
print("OmniServer started successfully")
+3
View File
@@ -1122,6 +1122,9 @@ async def _stage_worker_async(
vllm_config=vllm_config,
usage_context=usage_context,
engine_args=omni_engine_args,
disable_log_stats=bool(
engine_args.get("disable_log_stats", True) or getattr(omni_engine_args, "disable_log_stats", True)
),
)
omni_stage.set_async_engine(stage_engine)
if hasattr(omni_stage.async_engine, "log_stats") and omni_stage.async_engine.log_stats: