Prhub

#48276 [Bugfix][Test] Register Qwen/Qwen3.5-4B example model

原始 PR 作者 njhill 合并时间 2026-07-11 05:01 文件变更 1 提交数 1 评论 0 代码增减 +1 / -0

执行摘要

修复 Qwen3.5-4B 模型注册缺失导致测试失败

PR#48113 重定向了 DFlashDraftModel 的默认示例模型,移除了 tests/models/registry.py 中对 Qwen/Qwen3.5-4B 的唯一引用,导致 tests/renderers/test_hf.py::test_resolve_content_format_hf_defined[Qwen/Qwen3.5-4B-openai] 测试失败。该测试通过 HF_EXAMPLE_MODELS.find_hf_info 查找 Qwen/Qwen3.5-4B,但注册表中已无该模型,因此需要重新注册。

小型测试修复 PR,无需精读。但值得关注的是 _HfExamplesInfoextras 用法,它提供了一种在不改变默认示例模型的情况下注册额外模型的方式,可用于类似场景。

讨论亮点

无 review 评论。PR 由 robertgshaw2-redhat 直接批准合并。

实现拆解

tests/models/registry.py 中,Qwen3_5ForConditionalGeneration 条目的 _HfExamplesInfo 构造函数中新增 extras 参数,将 Qwen/Qwen3.5-4B 注册为额外模型。

  1. 定位注册条目:找到 tests/models/registry.pyQwen3_5ForConditionalGeneration 对应的 _HfExamplesInfo 调用,其默认模型为 Qwen/Qwen3.5-0.8B
  2. 添加 extras 参数:在原有参数后添加 extras={"4b": "Qwen/Qwen3.5-4B"},使得 find_hf_info("Qwen/Qwen3.5-4B") 能够解析到此模型的注册信息。
  3. 保持默认模型不变:未修改 Qwen/Qwen3.5-0.8B 这一默认值,保证现有兼容性。
文件 模块 状态 重要度
tests/models/registry.py 测试注册表 modified 3.81

关键源码片段

tests/models/registry.py test-coverage

唯一变更文件;在 Qwen3_5ForConditionalGeneration 注册项中添加 extras 参数,注册 Qwen/Qwen3.5-4B 作为额外示例模型。

# tests/models/registry.py 中 Qwen3_5ForConditionalGeneration 的示例模型注册条目
# 新增 extras 参数,将 Qwen/Qwen3.5-4B 注册为一个可查找的额外模型变体
"Qwen3_5ForConditionalGeneration": _HfExamplesInfo(
    "Qwen/Qwen3.5-0.8B", # 默认示例模型保持不变
    extras={"4b": "Qwen/Qwen3.5-4B"}, # 新增:额外注册 4B 变体
    max_model_len=4096,
),

评论区精华

没有提炼出高价值讨论线程

当前评论区没有形成足够清晰的争议点或结论,后续有更多讨论时会体现在这里。

风险与影响

本 PR 仅添加了一个注册条目,无任何逻辑变更,风险极低。不会影响生产代码或已有功能。

  • 测试:修复了 test_resolve_content_format_hf_defined 测试回归,确保使用 Qwen/Qwen3.5-4B 的测试和工具能正常工作。
  • 范围:仅影响测试基础设施中的模型注册表,不影响运行时模型加载或推理。

关联 Issue

未识别关联 Issue

当前没有检测到明确关联的 Issue 链接,后续同步到相关引用后会出现在这里。

完整报告

参与讨论