跳到内容

模型解析

vLLM 通过检查模型仓库中 config.json 文件的 architectures 字段，并查找已注册到 vLLM 的相应实现来加载 HuggingFace 兼容模型。然而，我们的模型解析可能会因以下原因失败：

  • 模型仓库的 config.json 缺少 architectures 字段。
  • 非官方仓库使用 vLLM 中未记录的其他名称来引用模型。
  • 同一个架构名称被多个模型使用，导致无法确定应该加载哪个模型。

要解决此问题，请通过将 config.json 覆盖项传递给 hf_overrides 选项来显式指定模型架构。例如：

from vllm import LLM

llm = LLM(
    model="cerebras/Cerebras-GPT-1.3B",
    hf_overrides={"architectures": ["GPT2LMHeadModel"]},  # GPT-2
)

我们的支持模型列表展示了 vLLM 可识别的模型架构。