2.1 vLLM-Omni + Wan2.1-T2V-1.3B测试数据

机器:

npu-smi info

复制代码
+------------------------------------------------------------------------------------------------+
| npu-smi 24.1.0                   Version: 24.1.0                                               |
+---------------------------+---------------+----------------------------------------------------+
| NPU   Name                | Health        | Power(W)    Temp(C)           Hugepages-Usage(page)|
| Chip                      | Bus-Id        | AICore(%)   Memory-Usage(MB)  HBM-Usage(MB)        |
+===========================+===============+====================================================+
| 0     910B3               | OK            | 95.2        43                0    / 0             |
| 0                         | 0000:C1:00.0  | 0           0    / 0          3874 / 65536         |
+===========================+===============+====================================================+
| 1     910B3               | OK            | 90.3        42                0    / 0             |
| 0                         | 0000:C2:00.0  | 0           0    / 0          50449/ 65536         |
+===========================+===============+====================================================+
| 2     910B3               | OK            | 90.6        43                0    / 0             |
| 0                         | 0000:81:00.0  | 0           0    / 0          3404 / 65536         |
+===========================+===============+====================================================+
| 3     910B3               | OK            | 94.1        44                0    / 0             |
| 0                         | 0000:82:00.0  | 0           0    / 0          3405 / 65536         |
+===========================+===============+====================================================+
| 4     910B3               | OK            | 96.6        47                0    / 0             |
| 0                         | 0000:01:00.0  | 0           0    / 0          44930/ 65536         |
+===========================+===============+====================================================+
| 5     910B3               | OK            | 92.0        44                0    / 0             |
| 0                         | 0000:02:00.0  | 0           0    / 0          7206 / 65536         |
+===========================+===============+====================================================+
| 6     910B3               | OK            | 100.3       46                0    / 0             |
| 0                         | 0000:41:00.0  | 0           0    / 0          3389 / 65536         |
+===========================+===============+====================================================+
| 7     910B3               | OK            | 95.5        47                0    / 0             |
| 0                         | 0000:42:00.0  | 0           0    / 0          59668/ 65536         |
+===========================+===============+====================================================+

单卡单实例:

复制代码
export ASCEND_RT_VISIBLE_DEVICES=2
export VLLM_WORKER_MULTIPROC_METHOD=spawn
source /usr/local/Ascend/ascend-toolkit/set_env.sh
source /usr/local/Ascend/nnal/atb/set_env.sh

vllm-omni serve /data/models/Wan2.1-T2V-1.3B-Diffusers \
 --omni --port 8023 --boundary-ratio 0.875 \
 --flow-shift 5.0 --cfg-parallel-size 2 --dtype float16

curl -X POST http://localhost:8023/v1/videos/sync \
 -F "prompt=A futuristic city at sunset" \
 -F "width=832" \
 -F "height=480" \
 -F "num_frames=81" \
 -F "fps=16" \
 -F "num_inference_steps=50" \
 -F "guidance_scale=4.0" \
 -F "seed=42" \
 -o /data/cjh/omini/server_test3.mp4

耗时6min+

双卡单实例:

复制代码
export ASCEND_RT_VISIBLE_DEVICES=2,3
export VLLM_WORKER_MULTIPROC_METHOD=spawn
export CFG_PARALLEL_SIZE=2
source /usr/local/Ascend/ascend-toolkit/set_env.sh
source /usr/local/Ascend/nnal/atb/set_env.sh

vllm-omni serve /data/models/Wan2.1-T2V-1.3B-Diffusers \
 --omni --port 8023 --boundary-ratio 0.875 \
 --flow-shift 5.0 --cfg-parallel-size 2 --dtype float16

curl -X POST http://localhost:8023/v1/videos/sync \
 -F "prompt=A futuristic city at sunset" \
 -F "width=832" \
 -F "height=480" \
 -F "num_frames=81" \
 -F "fps=16" \
 -F "num_inference_steps=50" \
 -F "guidance_scale=4.0" \
 -F "seed=42" \
 -o /data/cjh/omini/server_test3.mp4

耗时3min+

模型并行策略优化

复制代码
https://docs.vllm.com.cn/projects/vllm-omni/en/latest/user_guide/diffusion/parallelism_acceleration/#overview

模型并行测试结论:wan2.1
不支持
Ring / USP	长序列(长视频)	作用:减少单卡显存占用,序列越长加速越明显

支持
CFG-Parallel	固定2卡	速度×2,质量不变
作用:
	CFG引导分支并行
	扩散模型每步要跑两次(guidance分支 + 无guidance分支)
	CFG-Parallel 让两个分支同时跑在不同GPU上
	结果等价于串行50步,但时间减半

并发建议:

复制代码
请求并发数:AsyncOmniDiffusion initialized with model: ... batch_size: 1;
vLLM-Omni 每次只处理 1 个请求,batch_size=1 是单实例的上限。
这不是配置问题,是架构设计决定的Diffusion Worker 的调度模式不支持同时处理多个请求。

多实例+ Nginx 负载均衡
相关推荐
江厌019 分钟前
金融大模型私有化:信创合规下,算力该按哪个口径配
人工智能·深度学习·大模型·私有化部署·gpu·信创·ai服务器
颜安青18 分钟前
深度学习Pytorch下载推荐(笔记,仅记录学习)
pytorch·笔记·深度学习
qq_25294131681 小时前
牛目标检测数据集 | 牛只检测 智慧畜牧 动物识别 目标检测5017期
人工智能·深度学习·目标检测·计算机视觉·动物识别··牛识别
AI人工智能+1 小时前
基于深度学习的泰国文字识别系统,通过图像预处理、文字定位、整行序列预测、CNN特征提取、LSTM+CTC解码及语言模型校正六步,实现高精度泰文识别
深度学习·ocr·泰国文字识别
硅谷秋水1 小时前
Zero-WAM:基于人类视频的上下文世界-动作建模,用于开放式任务泛化
人工智能·深度学习·计算机视觉·语言模型·机器人·音视频
程序员于老七2 小时前
漫话大模型:先画轮廓再画细节——从 Flow Matching 看生成模型的“殊途同归“
深度学习·大模型·ai编程·diffusion·flowmatching
HYHYLLLL2 小时前
【2026年】多模态AI在水务场景的应用
人工智能·深度学习·计算机视觉·水务
CODER03042 小时前
ubuntu24.04:降内核+显卡驱动+cuda+cudnn+pytorch+anaconda+pycharm
人工智能·pytorch·深度学习·自然语言处理·cuda·ubuntu24.04·降内核
寒霜雨刃1 天前
【原创】海光Z100 DCU适配vLLM实录(二):W4A16大M Prefill结构重写、GDN融合与长上下文HIP Attention
深度学习·神经网络·amd·rocm·ai infra·海光z100·gfx906
IJCAST2 天前
IJCAST最新一期已经发布
人工智能·深度学习·神经网络