[CI/Build] VLM Test Consolidation (#9372)

Signed-off-by: Alex-Brooks <Alex.Brooks@ibm.com>
2024-10-30 10:32:17 -06:00
parent 211fe91aa8
commit cc98f1e079
38 changed files with 2381 additions and 3096 deletions
--- a/tests/models/encoder_decoder/vision_language/test_mllama.py
+++ b/tests/models/encoder_decoder/vision_language/test_mllama.py
@ -192,7 +192,7 @@ def _run_test(
            for prompts, images in inputs
        ]

-    def process(hf_inputs: BatchEncoding):
+    def process(hf_inputs: BatchEncoding, **kwargs):
        return hf_inputs

    with hf_runner(model,