Skip to content

Commit 3207427

Browse files
authored
[diffusion] CI: enable warmup as default (#19010)
1 parent d73f06f commit 3207427

File tree

5 files changed

+32
-103
lines changed

5 files changed

+32
-103
lines changed

python/sglang/multimodal_gen/test/scripts/gen_perf_baselines.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,8 +65,9 @@ def _build_server_extra_args(case: DiffusionTestCase) -> str:
6565
a += f" --ring-degree {server_args.ring_degree}"
6666
if server_args.lora_path:
6767
a += f" --lora-path {server_args.lora_path}"
68-
if server_args.warmup:
69-
a += " --warmup"
68+
69+
# default warmup
70+
a += " --warmup"
7071

7172
for extra_arg in server_args.extras:
7273
a += f" {extra_arg}"

python/sglang/multimodal_gen/test/server/ascend/testcase_configs_npu.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@
1212
DiffusionServerArgs(
1313
model_path="/root/.cache/modelscope/hub/models/Wan-AI/Wan2.1-T2V-1.3B-Diffusers",
1414
modality="video",
15-
warmup=True,
1615
custom_validator="video",
1716
),
1817
DiffusionSamplingParams(

python/sglang/multimodal_gen/test/server/perf_baselines.json

Lines changed: 19 additions & 44 deletions
Original file line numberDiff line numberDiff line change
@@ -523,28 +523,28 @@
523523
},
524524
"zimage_image_t2i": {
525525
"stages_ms": {
526-
"ConditioningStage": 0.01,
527-
"DecodingStage": 33.93,
528-
"TextEncodingStage": 409.74,
529526
"InputValidationStage": 0.03,
530-
"DenoisingStage": 808.69,
531-
"LatentPreparationStage": 0.14,
532-
"TimestepPreparationStage": 1.33
527+
"TextEncodingStage": 403.47,
528+
"ConditioningStage": 0.01,
529+
"TimestepPreparationStage": 1.41,
530+
"LatentPreparationStage": 0.11,
531+
"DenoisingStage": 756.21,
532+
"DecodingStage": 29.41
533533
},
534534
"denoise_step_ms": {
535-
"0": 23.4,
536-
"1": 28.26,
537-
"2": 100.72,
538-
"3": 100.48,
539-
"4": 99.62,
540-
"5": 99.83,
541-
"6": 99.67,
542-
"7": 99.3,
543-
"8": 100.83
544-
},
545-
"expected_e2e_ms": 1461.08,
546-
"expected_avg_denoise_ms": 89.36,
547-
"expected_median_denoise_ms": 99.67
535+
"0": 22.29,
536+
"1": 75.04,
537+
"2": 93.82,
538+
"3": 93.34,
539+
"4": 93.38,
540+
"5": 93.58,
541+
"6": 94.01,
542+
"7": 93.97,
543+
"8": 94.32
544+
},
545+
"expected_e2e_ms": 1192.92,
546+
"expected_avg_denoise_ms": 83.75,
547+
"expected_median_denoise_ms": 93.58
548548
},
549549
"zimage_image_t2i_multi_lora": {
550550
"stages_ms": {
@@ -596,31 +596,6 @@
596596
"expected_avg_denoise_ms": 144.49,
597597
"expected_median_denoise_ms": 156.16
598598
},
599-
"zimage_image_t2i_warmup": {
600-
"stages_ms": {
601-
"InputValidationStage": 0.02,
602-
"TextEncodingStage": 456.16,
603-
"ConditioningStage": 0.01,
604-
"TimestepPreparationStage": 0.98,
605-
"LatentPreparationStage": 0.06,
606-
"DenoisingStage": 889.42,
607-
"DecodingStage": 37.81
608-
},
609-
"denoise_step_ms": {
610-
"0": 16.49,
611-
"1": 94.63,
612-
"2": 109.65,
613-
"3": 110.05,
614-
"4": 109.39,
615-
"5": 110.58,
616-
"6": 109.52,
617-
"7": 110.54,
618-
"8": 115.24
619-
},
620-
"expected_e2e_ms": 1184.45,
621-
"expected_avg_denoise_ms": 98.46,
622-
"expected_median_denoise_ms": 109.65
623-
},
624599
"qwen_image_edit_ti2i": {
625600
"stages_ms": {
626601
"LatentPreparationStage": 0.16,

python/sglang/multimodal_gen/test/server/test_server_common.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -87,8 +87,8 @@ def diffusion_server(case: DiffusionTestCase) -> ServerContext:
8787
if server_args.lora_path:
8888
extra_args += f" --lora-path {server_args.lora_path}"
8989

90-
if server_args.warmup:
91-
extra_args += f" --warmup"
90+
# default warmup
91+
extra_args += f" --warmup"
9292

9393
for arg in server_args.extras:
9494
extra_args += f" {arg}"

0 commit comments

Comments
 (0)