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
Show 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
import pytest_asyncio
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"
......@@ -22,6 +22,7 @@ def server():
"--enforce-eager",
"--limit-mm-per-prompt",
json.dumps({"audio": 3, "video": 3}),
*ROCM_EXTRA_ARGS,
]
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