minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

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

Browse Source 
Co-authored-by: Alison Shao <alison.shao@MacBook-Pro-D2W773R9CD.local>
This commit is contained in:
Alison Shao
2026-04-10 17:49:56 -07:00
committed by GitHub
co-authored by Alison Shao
parent 0011d2aec0
commit 75223c5404
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/multimodal_gen/test/unit/test_transformer_quant.py
+5
View File
@@ -127,8 +127,13 @@ class TestTransformerQuantHelpers(unittest.TestCase):
@patch( @patch(
"sglang.multimodal_gen.runtime.loader.transformer_load_utils.get_metadata_from_safetensors_file" "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( def test_resolve_transformer_quant_load_spec_keeps_nunchaku_hook(
self, self,
_mock_download,
mock_metadata, mock_metadata,
_mock_quant_metadata, _mock_quant_metadata,
_mock_nvfp4, _mock_nvfp4,