cosmos-predict2

更新通知

Cosmos-Predict2 是物理 AI 环境 Cosmos World Foundation Models (WFMs) 的一个关键分支,专注于通过先进的世界模型进行未来状态预测。它提供了两项强大的功能:文本到图像生成,可通过文本描述生成高质量图像;以及视频到世界生成,可通过视频输入生成可视化模拟。下图展示了 Cosmos-Predict2 的架构。

资讯

• 2025-08-22: Cosmos-Predict2 现已在 PyPI 上推出! 请参阅 开始使用 了解使用方法。
• 2025-08-21: Cosmos-Predict2 现已包含预构建的依赖项! 请参阅 安装指南。
• 2025-08-15: 我们发布了 0.6B Text2Image 模型,支持快速标记器!
• 2025-07-10: 我们发布了 Predict2 + NATTEN,通过使用稀疏注意力实现了高达 2.6 倍的端到端推理加速 (视频)。
• 2025-06-11: 我们发布了训练后代码和推理代码,以及模型权重。如需代码演示,请参阅此 视频。

模型