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

Co-authored-by: Alison Shao <alison.shao@MacBook-Pro-D2W773R9CD.local>
2026-04-10 17:49:56 -07:00
parent 0011d2aec0
commit 75223c5404
1 changed files with 5 additions and 0 deletions
@@ -127,8 +127,13 @@ class TestTransformerQuantHelpers(unittest.TestCase):
    @patch(
        "sglang.multimodal_gen.runtime.loader.transformer_load_utils.get_metadata_from_safetensors_file"
    )
+    @patch(
+        "sglang.multimodal_gen.runtime.loader.transformer_load_utils.maybe_download_model",
+        side_effect=lambda path, **kw: path,
+    )
    def test_resolve_transformer_quant_load_spec_keeps_nunchaku_hook(
        self,
+        _mock_download,
        mock_metadata,
        _mock_quant_metadata,
        _mock_nvfp4,