Prhub

#22560 [Diffusion][CI] Fix nunchaku unit test broken by #22365

原始 PR 作者 alisonshao 合并时间 2026-04-11 08:49 文件变更 1 提交数 1 评论 6 代码增减 +5 / -0

执行摘要

修复因 #22365 引入的扩散模型单元测试失败,通过 Mock 下载函数避免本地路径被误识别为 HF 仓库。

修复由PR #22365引入的CI测试失败。PR body指出:#22365添加的_resolve_quant_config_from_transformer_override会调用maybe_download_model处理transformer权重路径,但测试中使用的本地路径/tmp/svdq-int4_r32.safetensors不存在于HF仓库,导致HF Hub验证失败,引发multimodal-gen-unit-test CI任务失败。

该PR变更简单,无需精读。对于关注扩散模型量化加载逻辑或CI测试稳定性的工程师,可快速浏览以了解Mock用法和测试修复模式。

讨论亮点

Review中仅有一名审核者(yhyang201)批准,无具体评论。从提交历史和PR body可推断,作者alisonshao在修复前已观察到CI失败(提供了失败链接),并通过Mock方式快速修复,未引发设计或实现上的深入讨论。

实现拆解

仅修改一个测试文件:python/sglang/multimodal_gen/test/unit/test_transformer_quant.py。在test_resolve_transformer_quant_load_spec_keeps_nunchaku_hook测试函数上添加一个新的@patch装饰器,Mock sglang.multimodal_gen.runtime.loader.transformer_load_utils.maybe_download_model函数,设置其side_effect为直接返回输入路径,避免实际下载操作。

文件 模块 状态 重要度
python/sglang/multimodal_gen/test/unit/test_transformer_quant.py multimodal_gen/test modified 3.0

关键符号

test_resolve_transformer_quant_load_spec_keeps_nunchaku_hook

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

评论区精华

测试修复策略 测试

无显式讨论,但从 PR body 和提交信息可推断,作者通过 Mock `maybe_download_model` 避免测试中本地路径被误下载。

结论:采用 Mock 方式修复,审核通过。 · 已解决

风险与影响

风险极低:

  1. 仅影响单元测试逻辑,不涉及生产代码。
  2. Mock策略简单直接,仅改变测试环境行为,不会引入回归。
  3. 变更范围极小(5行添加),易于验证。
    潜在风险:Mock可能掩盖了maybe_download_model在实际使用中的潜在问题,但鉴于这是针对测试假路径的修复,风险可控。

影响范围有限:

  1. 用户:无直接影响,纯测试修复。
  2. 系统:修复CI失败,确保multimodal-gen-unit-test任务通过,维护测试套件稳定性。
  3. 团队:减少CI噪音,提升开发效率。影响程度为低,仅涉及扩散模型量化相关的单元测试。
测试环境 Mock

关联 Issue

未识别关联 Issue

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

完整报告

参与讨论