vLLM首日跑通816B稀疏大模型:1M上下文只留四分之一KV缓存,TPU上单请求超960 token/秒
核心概要
vLLM在Muse Spark Open发布当天就提供了推理支持,这是一款总参数816B、每token激活42B的混合专家模型,支持1M上下文与原生图像、视频、音频输入。此前这类规模的模型往往要等社区适配数周;这次Nvidia与AMD镜像同步放出,并配套训练了DSpark投机解码草稿模型。TPU上配合投机解码,4颗v7芯片超过650 token/秒,8颗超过960 token/秒。
