wassname/vllm
1
0 Fork 0
mirror of https://github.com/wassname/vllm.git synced 2026-10-01 12:42:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
Signed-off-by: Alex-Brooks <Alex.Brooks@ibm.com>
This commit is contained in:
Alex Brooks authored and GitHub committed 2024-10-30 09:32:17 -07:00
1 parent 211fe91aa8
commit cc98f1e079
38 files changed
+2381 -3096
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

tests/models/decoder_only/audio_language/test_ultravox.py
+1 -1
View File
@@ -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