Skip to content

GitLab

"ssh:/git@developer.sourcefind.cn:2222/OpenDAS/vllm_cscc.git" did not exist on "e92d7085bfb25200c540391d0be06cb3b7c29ea4"
Commit 137e8a16 authored by 王敏's avatar 王敏
Browse files

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

parent 9736caa9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/lora/test_punica_variation.py
View file @ 137e8a16
...@@ -20,7 +20,7 @@ from vllm.utils import seed_everything ...@@ -20,7 +20,7 @@ from vllm.utils import seed_everything
from .utils import (generate_data, generate_data_for_expand_nslices, from .utils import (generate_data, generate_data_for_expand_nslices,
ref_torch_groupgemm) ref_torch_groupgemm)
HIDDEN_SIZES = [4097] HIDDEN_SIZES = [1024]
BATCHES = [1, 4, 16, 32] BATCHES = [1, 4, 16, 32]
NUM_LORA = [1, 8, 32, 128] 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