wangyuandHongsheng Liu
4e82192fd6
Signed-off-by: wangyu <410167048@qq.com> Signed-off-by: wangyu <53896905+yenuo26@users.noreply.github.com> Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>
108 lines
3.5 KiB
JSON
108 lines
3.5 KiB
JSON
[
|
|
{
|
|
"test_name": "test_wan22_i2v_single_device",
|
|
"description": "Single-device baseline",
|
|
"server_type": "vllm-omni",
|
|
"server_params": {
|
|
"model": "Wan-AI/Wan2.2-I2V-A14B-Diffusers",
|
|
"serve_args": {
|
|
"enable-diffusion-pipeline-profiler": true
|
|
}
|
|
},
|
|
"benchmark_params": [
|
|
{
|
|
"name": "832x480_frames81_steps4",
|
|
"dataset": "random",
|
|
"task": "i2v",
|
|
"num-prompts": 10,
|
|
"max-concurrency": 1,
|
|
"num-input-images": 1,
|
|
"seed": 42,
|
|
"enable-negative-prompt": true,
|
|
"random-request-config": [
|
|
{
|
|
"width": 832,
|
|
"height": 480,
|
|
"num_inference_steps": 4,
|
|
"num_frames": 81,
|
|
"fps": 16,
|
|
"weight": 1
|
|
}
|
|
],
|
|
"baseline": {
|
|
"throughput_qps": 0.04,
|
|
"latency_mean": 30,
|
|
"peak_memory_mb_mean": 75000
|
|
}
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"test_name": "test_wan22_i2v_usp2_vae_patch2_hsdp_slicing",
|
|
"description": "USP=2 + VAE patch parallel=2 + HSDP + VAE slicing",
|
|
"server_type": "vllm-omni",
|
|
"server_params": {
|
|
"model": "Wan-AI/Wan2.2-I2V-A14B-Diffusers",
|
|
"serve_args": {
|
|
"usp": 2,
|
|
"vae-patch-parallel-size": 2,
|
|
"use-hsdp": true,
|
|
"vae-use-slicing": true,
|
|
"enable-diffusion-pipeline-profiler": true
|
|
}
|
|
},
|
|
"benchmark_params": [
|
|
{
|
|
"name": "832x480_frames81_steps4",
|
|
"dataset": "random",
|
|
"task": "i2v",
|
|
"num-prompts": 10,
|
|
"max-concurrency": 1,
|
|
"num-input-images": 1,
|
|
"seed": 42,
|
|
"enable-negative-prompt": true,
|
|
"random-request-config": [
|
|
{
|
|
"width": 832,
|
|
"height": 480,
|
|
"num_inference_steps": 4,
|
|
"num_frames": 81,
|
|
"fps": 16,
|
|
"weight": 1
|
|
}
|
|
],
|
|
"baseline": {
|
|
"throughput_qps": 0.06,
|
|
"latency_mean": 20,
|
|
"peak_memory_mb_mean": 48000
|
|
}
|
|
},
|
|
{
|
|
"name": "1280x720_frames121_steps4",
|
|
"dataset": "random",
|
|
"task": "i2v",
|
|
"num-prompts": 10,
|
|
"max-concurrency": 1,
|
|
"num-input-images": 1,
|
|
"seed": 42,
|
|
"enable-negative-prompt": true,
|
|
"random-request-config": [
|
|
{
|
|
"width": 1280,
|
|
"height": 720,
|
|
"num_inference_steps": 4,
|
|
"num_frames": 121,
|
|
"fps": 16,
|
|
"weight": 1
|
|
}
|
|
],
|
|
"baseline": {
|
|
"throughput_qps": 0.01,
|
|
"latency_mean": 90,
|
|
"peak_memory_mb_mean": 56000
|
|
}
|
|
}
|
|
]
|
|
}
|
|
]
|