Commit 137e8a16 authored by 王敏's avatar 王敏
Browse files

[fix]修复单测test_punica_variation报共享内存不足的问题

parent 9736caa9
......@@ -20,7 +20,7 @@ from vllm.utils import seed_everything
from .utils import (generate_data, generate_data_for_expand_nslices,
ref_torch_groupgemm)
HIDDEN_SIZES = [4097]
HIDDEN_SIZES = [1024]
BATCHES = [1, 4, 16, 32]
NUM_LORA = [1, 8, 32, 128]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment