🟡 🤖 模型 发布于: · 1 分钟阅读 ·

vLLM:对TML Inkling的Day-0支持——万亿参数模型在GB200上达380 tokens/s

编辑插图:NVIDIA GB200 GPU上的token吞吐量图表,附vLLM和TML Inkling标志

vLLM为TML Inkling提供Day-0支持,这是一款拥有1万亿参数的多模态模型。在四块NVIDIA GB200 GPU上,系统通过MTP和长上下文推理实现每用户每秒最高380个tokens的吞吐量。

🤖

本文由人工智能基于一手来源生成。

什么是vLLM和TML Inkling?

vLLM大型语言模型推理开源引擎)发布了对TML Inkling的Day-0支持——这意味着新模型在发布当天即可获得支持,无需等待系统的后续更新。TML Inkling是一款拥有1万亿参数(10¹²)的多模态模型,属于公开可用模型中最重量级的一类。作为对比,GPT-4的参数量估计约为1.8万亿,而当前大多数开源模型的参数量在70亿至700亿之间。

性能表现如何?

四块NVIDIA GB200 GPU上,vLLM运行TML Inkling时每用户每秒可达380个tokens——对于这一规模的模型而言,这是极为出色的表现。系统采用MTPmulti-token prediction,多token预测——一次预测多个token而非逐个预测),进一步加速了文本和图像的生成。此外,该实现还支持长上下文推理以及多种并行方式(张量并行、流水线并行),可扩展至多块GPU。

这对AI基础设施开发意味着什么?

此次集成无需修改vLLM核心——TML Inkling可直接在现有系统上运行。这意味着已使用vLLM的机构可以立即开始部署这款万亿参数多模态模型,无需对现有技术栈进行额外适配。相比之前这类规模的模型需要专用推理框架或大量定制改造的情况,这一进展大幅降低了前沿模型生产部署的门槛。


来源:vLLM Blog

常见问题

AI模型的「Day-0支持」是什么意思?
Day-0支持意味着新模型在发布当天即可获得支持,无需等待推理引擎的后续更新。
什么是vLLM?
vLLM是一款用于大型语言模型推理的开源引擎,针对生产环境中的高吞吐量和低延迟进行了优化。

📬 AI 新闻直达您的邮箱

按您的方式定制每日摘要——自选主题、来源和频率,一键退订。