[CI][ROCm] Add LingBot nightly coverage (#7935)

Signed-off-by: haic0 <149741444+haic0@users.noreply.github.com>
Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
haic0andCursor authored and GitHub committed 2026-10-06 05:31:51 +00:00
1 parent 5e34e1c97d
commit 2bcd1e449a
2 files changed
+30 -4

No files matched your search

@@ -27,9 +27,10 @@ PROMPT = "a robotic arm picks up a red block"
NEGATIVE_PROMPT = "low quality, blurry, watermark, text"
SINGLE_CARD_FEATURE_MARKS = hardware_marks(res={"cuda": ["H100", "B200"]})
CFG_OFF_SINGLE_CARD_FEATURE_MARKS = hardware_marks(res={"cuda": ["H100", "B200"], "rocm": "MI325"})
def _get_diffusion_feature_cases(model: str):
def _get_diffusion_feature_cases(model: str, marks):
return [
pytest.param(
OmniServerParams(
@@ -37,12 +38,16 @@ def _get_diffusion_feature_cases(model: str):
server_args=["--model-class-name", "LingBotVideoPipeline"],
),
id="default",
marks=SINGLE_CARD_FEATURE_MARKS,
marks=marks,
),
]
@pytest.mark.parametrize("omni_server", _get_diffusion_feature_cases(MODEL), indirect=True)
@pytest.mark.parametrize(
"omni_server",
_get_diffusion_feature_cases(MODEL, CFG_OFF_SINGLE_CARD_FEATURE_MARKS),
indirect=True,
)
def test_cfg_off(omni_server: OmniServer, openai_client: OpenAIClientHandler) -> None:
request_config = {
"model": omni_server.model,
@@ -62,7 +67,11 @@ def test_cfg_off(omni_server: OmniServer, openai_client: OpenAIClientHandler) ->
openai_client.send_video_diffusion_request(request_config)
@pytest.mark.parametrize("omni_server", _get_diffusion_feature_cases(MODEL), indirect=True)
@pytest.mark.parametrize(
"omni_server",
_get_diffusion_feature_cases(MODEL, SINGLE_CARD_FEATURE_MARKS),
indirect=True,
)
def test_batch_cfg_extra_params(omni_server: OmniServer, openai_client: OpenAIClientHandler) -> None:
request_config = {
"model": omni_server.model,