vllm-ascend支持的模型

模型与算力

讲清 vLLM 昇腾版支持哪些模型、怎么自己核对清单,避开选型时把「跑得起来」当成「跑得好」的坑。

你在昇腾卡上要挑一个模型上线,翻文档只看到一句「支持主流模型」,具体到你要用的那个能不能跑、要不要改配置、多模态行不行,全靠猜。这篇不讲排行榜,只给一套核对方法:从哪查清单、按什么维度筛、什么情况下这份清单不适用。

你在昇腾卡上要挑一个模型上线,翻文档只看到一句「支持主流模型」,具体到你要用的那个能不能跑、要不要改配置、多模态行不行,全靠猜。这篇文章不讲排行榜,只给一套核对方法:从哪查清单、按什么维度筛、什么情况下这份清单不适用。

先说清楚一件事:本文里的「支持」不是拍板结论,而是一个需要你自己动手验证的状态。任何清单都会随框架版本、驱动版本、量化格式变化,所以真正有用的不是记住某个模型名,而是学会怎么在站内把当时的清单和硬件条件对齐。

先把「支持」这两个字拆开

很多人把「权重能加载」当成「支持」,结果跑起来发现输出乱码、速度慢到不能用。选型时先给自己分三层:第一层是能加载,第二层是能正常出结果,第三层是性能和精度都达到上线标准。大部分踩坑都发生在把第一层当成第三层。

能加载
模型权重读进显存、进程不报错,但不保证输出质量
能出结果
给定输入能返回语义正确的内容,精度可接受
能上线
吞吐、延迟、并发、精度都满足你的业务指标
  • 对照站内模型库和加速卡规格库,确认参数量、量化格式与显存对得上,不要只看模型名;
  • 昇腾侧常见限制集中在自定义算子、量化格式和并行方式,选型时优先挑社区里已有人跑过的结构;
  • 如果你的模型带自定义注意力或特殊位置编码,先按「可能不支持」预期排期;
  • 判断标准很简单:核对清单时任何一项对不上,就当作需要额外适配工作,别默认它会自动兼容。

自己核清单的四步动作

下面这套顺序是让你从零到有一个可复现的结论,不是让你背清单。每一步都有验证动作,做完能拿到一条可回滚的基线。

  1. 找对清单来源在 vLLM 昇腾分支的文档与模型支持页找到当前版本列出的模型。注意别拿通用版清单套用,两者的支持范围经常不一致。
  2. 算硬件账到站内部署测算页按参数量和量化档位算一遍显存与并发,确认卡型和卡数够不够。这一步能提前排除「模型支持但机器跑不动」的情况。
  3. 最小输入跑通用一句最短的提示词跑一次生成,重点看加载日志里有没有算子回退或权重裁剪告警。有告警就说明精度可能打折。
  4. 记下版本组合把这次跑通的框架版本、卡型、量化格式写下来,作为以后升级或换模型的对比基线。没有基线就没法判断升级是否引入退化。

这套流程也有不适用的时候:如果你只是做一次演示、随时能换云上 API,没必要走完整验证;如果模型已经有人在同一卡型上跑通并给出可复现记录,你可以直接从第三步开始。是否值得投入,取决于这个模型会不会进生产。

选型时真正该看的几个维度

支持清单只回答「有没有」,选型还要回答「划不划算」。下面几个维度比模型名更能决定你后面会不会返工。

维度要确认什么去哪里核对
任务类型纯文本生成、长上下文、多模态理解走的是不同支持路径站内模型库对应分类页
量化格式某些格式在昇腾上支持更成熟,换格式常比换模型省事加速卡规格库与部署测算页
并发与吞吐写清楚预期并发,再确认卡型和卡数能撑住站内算力与部署测算相关页面
许可证能部署不等于能商用,授权边界单独查模型库里的许可说明字段
  • 先定任务类型,再挑模型,顺序反了容易选到不匹配的结构;
  • 量化格式的优先级往往高于模型名字,同一模型换格式可能直接可用;
  • 把并发预期写下来再去核对硬件,别等压测才发现只能单路跑;
  • 许可证和商用边界单独查一遍,这一步省不得。

清单没写你的模型,怎么办

清单里找不到你的模型很常见,不代表这条路走不通,但要先判断缺口的性质,再决定投入多少。判断顺序是:先看是不是整体缺,再看是不是只差某个配置项。

缺口类型典型表现处理优先级
整体不支持模型结构或算子完全没进入支持范围优先换模型或换路线
配置级缺口模型在列表里,但你用的量化格式或并行方式不在换格式或换并行配置
边缘缺口只在特定输入长度或多模态分支出问题评估适配工作量后再决定
  • 能降级就降级:换同系列稠密版本、换更通用的量化格式,通常比啃适配划算;
  • 真要自己适配,先估三块工作量:算子改写、精度对齐、回归测试;
  • 给出退出条件:如果几天内跑不到可接受的精度和速度,就换模型或换回通用 GPU 路线,别无限期投入;
  • 把这次适配的结论记回站内笔记,下次别人遇到同样问题能省一轮试错。

需要提醒的是,上面这些判断都基于你当前拿到的清单和硬件。具体到参数量、显存占用、吞吐数字,站内页面会随时间和版本更新,务必到站内对应页面核对当时的数字,不要凭记忆下结论。

下一步打开站内模型库和部署测算页,把你候选的两三个模型按今天的版本各核一遍支持状态和硬件账,得到一份能直接拿去评审的对照表。

常见问题

vLLM 昇腾版是不是跟通用版支持一样的模型?

不一样。两边支持范围有交集但不重合,昇腾侧的清单受算子实现和量化格式限制,通常比通用版窄。正确做法是以昇腾分支自己的支持列表为准,再把你的模型和量化格式单独核一遍,不要直接拿通用版的结论套过来。

模型在清单里,是不是就能直接上生产?

不一定。清单只说明能跑,不代表吞吐、延迟和精度达到你业务要求。上线前至少跑一遍自己的真实输入,量出首 token 延迟、生成速度和并发表现,再决定要不要调量化档位或加卡。这些数字建议到站内的测速与算力页面拿基线对照。

多模态模型在昇腾上支持情况怎么判断?

比纯文本更保守,视觉编码器和图像预处理环节常是短板。判断顺序是:先确认该多模态结构有没有被列进支持范围,再看图像侧是否走的是标准实现,最后用几张真实图片跑通看输出是否正常。拿不准就先按不支持预期排期。

本文依据各平台公开文档与官方公告整理,只做中文归纳;具体做法、参数与限制以官方文档为准。