Skip to content

GitLab

Unverified Commit 886ccbe5 authored by Zhuohan Li's avatar Zhuohan Li Committed by GitHub
Browse files

[CI/Build] Reduce the number of redundant cases to test for LoRA (#24276) 


Signed-off-by: default avatarZhuohan Li <zhuohan123@gmail.com>
parent adc3ddb4
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/lora/test_layers.py
View file @ 886ccbe5
......@@ -60,9 +60,9 @@ DEVICES = ([
# prefill stage(True) or decode stage(False)
STAGES = [True, False]
NUM_RANDOM_SEEDS = 6
NUM_RANDOM_SEEDS = 2
VOCAB_PARALLEL_EMBEDDING_TEST_NUM_RANDOM_SEEDS = 128
VOCAB_PARALLEL_EMBEDDING_TEST_NUM_RANDOM_SEEDS = 2
@pytest.fixture(autouse=True)
......
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