Skip to content

GitLab

Unverified Commit 8b632575 authored by Andreas Karatzas's avatar Andreas Karatzas Committed by GitHub
Browse files

[ROCm][CI] Add ROCM_EXTRA_ARGS to audio_in_video test server fixture (#37349) 


Signed-off-by: default avatarAndreas Karatzas <akaratza@amd.com>
parent a0dd1995
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
tests/entrypoints/openai/chat_completion/test_audio_in_video.py
View file @ 8b632575
...@@ -9,7 +9,7 @@ import pytest ...@@ -9,7 +9,7 @@ import pytest
import pytest_asyncio import pytest_asyncio
from tests.conftest import VideoTestAssets from tests.conftest import VideoTestAssets
from tests.utils import RemoteOpenAIServer from tests.utils import ROCM_EXTRA_ARGS, RemoteOpenAIServer
MODEL_NAME = "Qwen/Qwen2.5-Omni-3B" MODEL_NAME = "Qwen/Qwen2.5-Omni-3B"
...@@ -22,6 +22,7 @@ def server(): ...@@ -22,6 +22,7 @@ def server():
"--enforce-eager", "--enforce-eager",
"--limit-mm-per-prompt", "--limit-mm-per-prompt",
json.dumps({"audio": 3, "video": 3}), json.dumps({"audio": 3, "video": 3}),
*ROCM_EXTRA_ARGS,
] ]
with RemoteOpenAIServer( with RemoteOpenAIServer(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment