From 17192c9a0e78098041860994a6578fbe387f71b4 Mon Sep 17 00:00:00 2001 From: drbh Date: Fri, 17 Jan 2025 16:19:02 +0000 Subject: [PATCH] fix: remove test debug params --- integration-tests/models/test_flash_qwen2_vl_video.py | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/integration-tests/models/test_flash_qwen2_vl_video.py b/integration-tests/models/test_flash_qwen2_vl_video.py index 79eea3c7..3be67f07 100644 --- a/integration-tests/models/test_flash_qwen2_vl_video.py +++ b/integration-tests/models/test_flash_qwen2_vl_video.py @@ -5,13 +5,7 @@ import requests @pytest.fixture(scope="module") def qwen2_vl_handle(launcher): - with launcher( - "Qwen/Qwen2-VL-7B-Instruct", - max_input_length=10_000, - max_batch_prefill_tokens=10_000, - max_total_tokens=10_001, - cuda_graphs=[0], - ) as handle: + with launcher("Qwen/Qwen2-VL-7B-Instruct") as handle: yield handle