diff --git a/tests/models/multimodal/generation/test_memory_leak.py b/tests/models/multimodal/generation/test_memory_leak.py index 743a71f928..5ee505257c 100644 --- a/tests/models/multimodal/generation/test_memory_leak.py +++ b/tests/models/multimodal/generation/test_memory_leak.py @@ -25,7 +25,7 @@ TEST_IMAGE_NAMES = [ ] MAX_MODEL_LEN = 8192 REQUESTS_PER_ROUND = 4 -WARMUP_ROUNDS = 1 +WARMUP_ROUNDS = 2 MEASURED_ROUNDS = 16 GPU_GROWTH_THRESHOLD_MIB = 0 CPU_PEAK_GROWTH_THRESHOLD_MIB = 0