[ROCm][CI] Mark gemma3 as large GPU test to avoid OOM on MI250 (#37610)

Signed-off-by: Andreas Karatzas <akaratza@amd.com>
2026-03-20 23:57:25 -05:00
parent 1fa1e53a73
commit 0d50fa1db6
2 changed files with 19 additions and 15 deletions
--- a/tests/models/multimodal/generation/test_common.py
+++ b/tests/models/multimodal/generation/test_common.py
@@ -220,7 +220,10 @@ VLM_TEST_SETTINGS = {
        vllm_runner_kwargs={
            "model_impl": "transformers",
        },
-        marks=[pytest.mark.core_model],
+        marks=[
+            pytest.mark.core_model,
+            *([large_gpu_mark(min_gb=80)] if current_platform.is_rocm() else []),
+        ],
    ),
    "idefics3-transformers": VLMTestInfo(
        models=["HuggingFaceTB/SmolVLM-256M-Instruct"],