执行摘要

• 一句话：修复 Mamba2 无 architectures 字段加载崩溃
• 推荐动作：值得合入，改动小而精准，修复了一个明确的用户可见崩溃。

功能与动机

HF 官方 Mamba2 转换权重（如 AntonV/mamba2-130m-hf）的 config.json 省略了 architectures 字段，导致 vLLM 引擎初始化时因 mamba_block_size 为 None 而断言失败。

实现拆解

1. 在 vllm/config/vllm.py 的 try_verify_and_update_config 方法中，引入 ModelRegistry._normalize_arch 导入。
2. 当 MODELS_CONFIG_MAP.get(architecture) 返回 None 时，用 ModelRegistry._normalize_arch(architecture, self.model_config) 对架构名归一化（例如将 'Mamba2Model' 转为 'Mamba2ForCausalLM'），然后重新查找。
3. 若重查找成功，则调用对应的 verify_and_update_config，确保 mamba_block_size 等配置被正确设置。

关键文件：

• vllm/config/vllm.py（模块 配置层；类别 source；类型 core-logic）: 核心修复文件：在 try_verify_and_update_config 中添加架构名归一化降级逻辑。

关键符号：try_verify_and_update_config

关键源码片段

vllm/config/vllm.py

核心修复文件：在 try_verify_and_update_config 中添加架构名归一化降级逻辑。

# vllm/config/vllm.py - try_verify_and_update_config 方法片段
​
def try_verify_and_update_config(self):
    if self.model_config is None:
        return
​
    # 避免重复执行
    if getattr(self.model_config, "config_updated", False):
        return
    self.model_config.config_updated = True
​
    architecture = self.model_config.architecture
    if architecture is None:
        return
​
    from vllm.model_executor.models import ModelRegistry
    from vllm.model_executor.models.config import (
        MODELS_CONFIG_MAP,
        HybridAttentionMambaModelConfig,
    )
​
    cls = MODELS_CONFIG_MAP.get(architecture, None)
    if cls is None:
        # `architecture` 可能是 HF 基础模型名（如 "Mamba2Model"，
        # 当 `architectures` 字段缺失时由 vLLM 注入），需要归一化
        # 为可运行的架构名（如 "Mamba2ForCausalLM"），以避免
        # per-arch 配置钩子被跳过。
        architecture = ModelRegistry._normalize_arch(
            architecture, self.model_config
        )
        cls = MODELS_CONFIG_MAP.get(architecture, None)
    if cls is not None:
        cls.verify_and_update_config(self)
​
    # 后续的混合注意力 / 分类等配置处理不变 ...

评论区精华

审核者 yewentao256 最初建议添加单元测试，但后续认为改动太小无需专门测试。之后询问测试是否已被移除，作者确认 PR 仍需要，测试已被移除。最终审核者批准。

• 是否需要单元测试 (testing): 一致认为无需专用测试，现有测试覆盖足够。

风险与影响

• 风险：变更仅发生在架构名直接查找失败后的降级路径，对已有 architectures 字段的正常 checkpoint 无影响。风险极低，但理论上若 normalize_arch 抛出异常，则加载失败。
• 影响：影响范围小：仅影响缺少 architectures 字段的 Mamba/Mamba2 checkpoint 用户。修复后这类 checkpoint 可以成功加载为 embedding 模型。
• 风险标记：边缘路径安全降级

关联脉络

• 暂无明显关联 PR