c139ee4aed
Signed-off-by: Huang, Zeyu <11222265+fhfuih@users.noreply.github.com> Signed-off-by: Didan Deng <33117903+wtomin@users.noreply.github.com> Co-authored-by: Gao Han <hgaoaf@connect.ust.hk> Co-authored-by: Alicia <115451386+congw729@users.noreply.github.com> Co-authored-by: Didan Deng <33117903+wtomin@users.noreply.github.com>
158 lines
5.2 KiB
JSON
158 lines
5.2 KiB
JSON
[
|
|
{
|
|
"test_name": "test_qwen_image_single_device",
|
|
"description": "Single-device baseline (no parallelism)",
|
|
"server_type": "vllm-omni",
|
|
"server_params": {
|
|
"model": "Qwen/Qwen-Image",
|
|
"serve_args": {
|
|
"enable-diffusion-pipeline-profiler": true
|
|
}
|
|
},
|
|
"benchmark_params": [
|
|
{
|
|
"name": "512x512_steps20",
|
|
"dataset": "random",
|
|
"task": "t2i",
|
|
"width": 512,
|
|
"height": 512,
|
|
"num-inference-steps": 20,
|
|
"num-prompts": 10,
|
|
"max-concurrency": 1,
|
|
"enable-negative-prompt": true,
|
|
"baseline": {
|
|
"throughput_qps": 0.30,
|
|
"latency_mean": 3.50,
|
|
"peak_memory_mb_mean": 67000
|
|
}
|
|
},
|
|
{
|
|
"name": "1536x1536_steps35",
|
|
"dataset": "random",
|
|
"task": "t2i",
|
|
"width": 1536,
|
|
"height": 1536,
|
|
"num-inference-steps": 35,
|
|
"num-prompts": 10,
|
|
"max-concurrency": 1,
|
|
"enable-negative-prompt": true,
|
|
"baseline": {
|
|
"throughput_qps": 0.037,
|
|
"latency_mean": 27.0,
|
|
"peak_memory_mb_mean": 74000
|
|
}
|
|
}
|
|
]
|
|
},
|
|
|
|
{
|
|
"test_name": "test_qwen_image_ulysses2_cfg2_vae_patch4",
|
|
"description": "Ulysses SP=2 + CFG-parallel=2 + VAE Patch Parallel=4",
|
|
"server_type": "vllm-omni",
|
|
"server_params": {
|
|
"model": "Qwen/Qwen-Image",
|
|
"serve_args": {
|
|
"ulysses-degree": 2,
|
|
"cfg-parallel-size": 2,
|
|
"vae-patch-parallel-size": 4,
|
|
"vae-use-tiling": true,
|
|
"enable-diffusion-pipeline-profiler": true
|
|
}
|
|
},
|
|
"benchmark_params": [
|
|
{
|
|
"name": "512x512_steps20",
|
|
"dataset": "random",
|
|
"task": "t2i",
|
|
"width": 512,
|
|
"height": 512,
|
|
"num-inference-steps": 20,
|
|
"num-prompts": 10,
|
|
"max-concurrency": 1,
|
|
"enable-negative-prompt": true,
|
|
"baseline": {
|
|
"throughput_qps": 0.1,
|
|
"latency_mean": 2.34,
|
|
"peak_memory_mb_mean": 61000
|
|
}
|
|
},
|
|
{
|
|
"name": "1536x1536_steps35",
|
|
"dataset": "random",
|
|
"task": "t2i",
|
|
"width": 1536,
|
|
"height": 1536,
|
|
"num-inference-steps": 35,
|
|
"num-prompts": 10,
|
|
"max-concurrency": 1,
|
|
"enable-negative-prompt": true,
|
|
"baseline": {
|
|
"throughput_qps": 0.1,
|
|
"latency_mean": 9.1,
|
|
"peak_memory_mb_mean": 61000
|
|
}
|
|
}
|
|
]
|
|
},
|
|
|
|
{
|
|
"test_name": "test_qwen_image_ulysses2_cfg2_cache_dit",
|
|
"description": "Ulysses SP=2 + CFG-parallel=2 + CacheDiT acceleration",
|
|
"server_type": "vllm-omni",
|
|
"server_params": {
|
|
"model": "Qwen/Qwen-Image",
|
|
"serve_args": {
|
|
"ulysses-degree": 2,
|
|
"cfg-parallel-size": 2,
|
|
"cache-backend": "cache_dit",
|
|
"cache-config": {
|
|
"Fn_compute_blocks": 1,
|
|
"Bn_compute_blocks": 0,
|
|
"max_warmup_steps": 4,
|
|
"residual_diff_threshold": 0.24,
|
|
"max_continuous_cached_steps": 3,
|
|
"enable_taylorseer": false,
|
|
"taylorseer_order": 1,
|
|
"scm_steps_mask_policy": null,
|
|
"scm_steps_policy": "dynamic"
|
|
},
|
|
"enable-diffusion-pipeline-profiler": true
|
|
}
|
|
},
|
|
"benchmark_params": [
|
|
{
|
|
"name": "512x512_steps20",
|
|
"dataset": "random",
|
|
"task": "t2i",
|
|
"width": 512,
|
|
"height": 512,
|
|
"num-inference-steps": 20,
|
|
"num-prompts": 10,
|
|
"max-concurrency": 1,
|
|
"enable-negative-prompt": true,
|
|
"baseline": {
|
|
"throughput_qps": 0.1,
|
|
"latency_mean": 3.50,
|
|
"peak_memory_mb_mean": 67000
|
|
}
|
|
},
|
|
{
|
|
"name": "1536x1536_steps35",
|
|
"dataset": "random",
|
|
"task": "t2i",
|
|
"width": 1536,
|
|
"height": 1536,
|
|
"num-inference-steps": 35,
|
|
"num-prompts": 10,
|
|
"max-concurrency": 1,
|
|
"enable-negative-prompt": true,
|
|
"baseline": {
|
|
"throughput_qps": 0.1,
|
|
"latency_mean": 6.15,
|
|
"peak_memory_mb_mean": 74000
|
|
}
|
|
}
|
|
]
|
|
}
|
|
]
|