[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/decoder_only/audio_language/test_ultravox.py
+++ b/tests/models/decoder_only/audio_language/test_ultravox.py
@ -92,7 +92,7 @@ def run_test(
            for vllm_prompt, _, audio in prompts_and_audios
        ]

-    def process(hf_inputs: BatchEncoding):
+    def process(hf_inputs: BatchEncoding, **kwargs):
        hf_inputs["audio_values"] = hf_inputs["audio_values"] \
            .to(torch_dtype)  # type: ignore
        return hf_inputs