Reduce test memory

Signed-off-by: Jee Jee Li <[email protected]>
vllm-project · Jan 12, 2025 · d8cb9af · d8cb9af
1 parent e208468
commit d8cb9af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/lora/test_punica_ops_variation.py b/tests/lora/test_punica_ops_variation.py
@@ -20,7 +20,7 @@
                     generate_data_for_expand_nslices,
                     generate_data_for_nslices)
 
-HIDDEN_SIZES = [4097]
+HIDDEN_SIZES = [2049]
 
 BATCHES = [1, 4, 16, 32]
 NUM_LORA = [1, 8, 32, 128]