[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:
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,
|
||||
|
||||
Reference in new issue
Block a user