全景路径

vLLM-Omni 部署在 AWS SageMaker AI

Deploy two generative media models from one AWS vLLM-Omni Deep Learning Container on Amazon SageMaker AI. Generate an image with FLUX.2-klein through real-time inference, then anim

返回指南列表

# vLLM-Omni 部署在 AWS SageMaker AI

你正在寻找从一个容器同时运行生成图像和视频模型的路径吗?vLLM-Omni 配合 AWS SageMaker AI 可以完成这项任务:通过实时推理调用 FLUX.2-klein 生成图片,再用图像条件动画生成视频。这对开发者、内容创作者和需要多模态 AI 推理的团队特别适用。

适用场景:如果你在 AWS 上部署生产级生成式应用,需要低延迟图像输出和高时长视频任务,却不想管理多个容器镜像,vLLM-Omni DLC 就是最优选择。它复用同一个基础镜像,通过路由中间件实现高效分发。

决策依据:直接看官方部署示例(2026-09-28 发布),确认 GPU 实例类型和异步推理模式匹配你的负载。RooHome 提供全景路径分流后,你可以根据意图跳转独立垂直站获取更多数据。

为什么选择 vLLM-Omni + SageMaker

vLLM-Omni 扩展了 vLLM,支持文本、图像、视频等多模态通过 OpenAI 兼容 API 提供服务。AWS SageMaker AI 则提供托管环境、自动扩展和计费监控。核心优势是单容器多模型:一个 DLC 镜像可部署实时端点(图像)和异步端点(视频),节省部署和维护成本。

核心架构

同一个容器加载不同模型:

  • 实时端点:运行 FLUX.2-klein,路由到 /v1/images/generations(响应几秒内返回 base64 PNG)。
  • 异步端点:运行 Wan2.1-VACE,路由到 /v1/videos/sync(通过 S3 上传/下载 MP4)。

请求流程:CLI 或 Streamlit 应用先调用图像端点,处理 PNG 后上传到 S3,再提交视频请求,异步完成。

准备工作

1. AWS 账户 + 已配置凭证。

2. SageMaker 执行角色(需 S3 权限和创建端点权限)。

3. Git、Python 3.11+。

4. 区域配额(ml.g6.xlarge / ml.g6e.xlarge)。

参考官方定价页面: 以估算 GPU 小时成本。

部署步骤

1. 克隆示例仓库:


 git clone 

 cd sagemaker-genai-hosting-examples/03-features/vllm-omni-image-video

2. 创建虚拟环境并安装依赖:


 python -m venv .venv

 source .venv/bin/activate

 pip install -r requirements.txt

3. 部署端点(设置 SM_VLLM_MODEL 环境变量):


 export SAGEMAKER_ROLE_ARN=你的角色 ARN

 python deploy.py --role-arn "$SAGEMAKER_ROLE_ARN" --region us-east-1

脚本会创建实时图像端点和异步视频端点,保存状态到 .vllm_omni_media_state.json。

4. 执行生成命令:


 python generate.py --image-prompt "Cinematic photograph of a coastal observatory at sunrise" --video-prompt "Slow camera push-in toward the coastal observatory; preserve the building and coastline"

图像返回后,脚本自动处理、上传、异步调用并保存 MP4 到 outputs/ 目录。

5. 启动 Streamlit 界面(可选):


 streamlit run app.py

在界面中调整提示词、尺寸和种子,直接生成并下载图像 + 视频。

6. 清理资源:


 python cleanup.py

完整代码和部署脚本请参阅 AWS 官方示例仓库:

成本与性能参考

项目 实时图像端点 异步视频端点
实例类型 ml.g6.xlarge ml.g6e.xlarge
启动时间 ~9.5 分钟 ~8.7 分钟
图像延迟 4.7 秒(warm) —
视频模型延迟 — 8.9 秒
总获取 MP4 — ~11.8 秒

数据为单次运行参考值,实际以官方 SageMaker 定价页为准:

延伸阅读

  • [vLLM-Omni 文档](官方配置和环境变量说明)
  • [AWS SageMaker 异步推理](深入了解 S3 流程)
  • [RooHome 价格雷达](查看同类多模态模型定价对比)
  • [RooHome 官方 API 文档](查看兼容 API 用法)
  • [RooHome API 中转指南](集成测试参考)

风险与边界

不要尝试:非官方修改器或绕过支付机制,这些可能导致账号封禁、费用失控或法律风险。vLLM-Omni 官方仅支持授权的 Hugging Face 模型和 AWS 认证。

常见误判:以为异步模式适合所有视频任务;实际大批量生产可能仍需优化 VAE 切片和实例类型。升级 DLC 或模型版本后,务必测试端点稳定性——升级后必挂风险高。

边界:此部署仅适用于 GPU 实例,CPU 模式不支持多模态。输出 MP4 需自行验证版权和内容安全。以上为技术参考,非法律意见,实际操作请以 AWS 官方文档和 SageMaker 控制台为准,费用实时核对:

English summary

This guide explains how to deploy vLLM-Omni on AWS SageMaker AI to generate images with FLUX.2-klein and animate them into videos using Wan2.1-VACE. It covers prerequisites, the two-step endpoint deployment (real-time image, async video), the full request flow via S3, and clean-up steps. The sample uses the pinned omni-sagemaker-cuda DLC for single-container efficiency. Performance data is from a US-East-1 run; always check current SageMaker pricing. This is not legal advice—review AWS terms for model usage and billing. Perfect for developers needing fast image-to-video pipelines on AWS infrastructure.