Prhub

#26522 [NemotronH] Fix weight-loading unit test broken by Puzzle support

原始 PR 作者 Fridge003 合并时间 2026-05-28 08:44 文件变更 1 提交数 1 评论 3 代码增减 +1 / -1

执行摘要

修复 NemotronH 权重加载测试因配置键缺失而失败

修复 CI 中 test_nemotron_h_weight_loading.py 的失败,原因是 load_weights 需要 config.max_n_routed_experts 但伪造配置未提供该属性。

合并即可,无需精读。

讨论亮点

无讨论。

实现拆解

将测试文件 test/registered/unit/models/test_nemotron_h_weight_loading.py_make_minimal_model 方法的第 38 行从 SimpleNamespace(n_routed_experts=2) 改为 SimpleNamespace(n_routed_experts=2, max_n_routed_experts=2),使伪造配置与真实 NemotronHConfig 的行为对齐。

文件 模块 状态 重要度
test/registered/unit/models/test_nemotron_h_weight_loading.py 测试 modified 4.03

关键符号

_make_minimal_model

分析完成后,这里会展示 LLM 生成的相对完整源码片段和详细注释。

评论区精华

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

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

风险与影响

低风险:仅修改测试夹具,生产代码不变。

影响范围仅限于修复该单元测试在 CI 中的失败,无其他影响。

关联 Issue

未识别关联 Issue

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

完整报告

参与讨论