Skip to content

Commit e371c10

Browse files
committed
reduce the size of the test
Signed-off-by: Xiongfei Wei <isaacwxf23@gmail.com>
1 parent ab09b7c commit e371c10

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/lora/test_layers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -198,7 +198,7 @@ def create_random_inputs(
198198

199199

200200
@torch.inference_mode()
201-
@pytest.mark.parametrize("num_loras", [1, 2, 4, 9])
201+
@pytest.mark.parametrize("num_loras", [1, 4, 9])
202202
@pytest.mark.parametrize("repeats", [2])
203203
@pytest.mark.parametrize("stage", [True, False])
204204
def test_column_parallel_packed(dist_init, num_loras, repeats, stage) -> None:

0 commit comments

Comments
 (0)