From 96e8a778551ed6572a1586bd2e04d9abd165863c Mon Sep 17 00:00:00 2001 From: Aurick Qiao Date: Wed, 23 Oct 2024 19:58:00 +0000 Subject: [PATCH] fix --- tests/lora/test_minicpmv.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/lora/test_minicpmv.py b/tests/lora/test_minicpmv.py index cdde90d5cad39..0ca97c7469eda 100644 --- a/tests/lora/test_minicpmv.py +++ b/tests/lora/test_minicpmv.py @@ -61,7 +61,7 @@ def test_minicpmv_lora(minicpmv_lora_files): max_loras=4, max_lora_rank=64, trust_remote_code=True, - gpu_memory_utilization=0.97 # This model is pretty big for CI gpus + gpu_memory_utilization=0.97, # This model is pretty big for CI gpus enable_chunked_prefill=True, )