Skip to content

Commit 8fc1908

Browse files
committed
fix function name
Signed-off-by: ZX-ModelCloud <zx@modelcloud.ai>
1 parent d60f3b4 commit 8fc1908

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

tests/test_convert_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
# Contact: qubitium@modelcloud.ai, x.com/qubitium
55
import torch
66

7-
from defuser.modeling.fused_moe.replace_modules import materialize_model_
7+
from defuser.modeling.fused_moe.replace_modules import materialize_model
88
from transformers import AutoModelForCausalLM, AutoModelForImageTextToText
99

1010
from defuser import convert_hf_model
@@ -45,7 +45,7 @@ def test_qwen3_5_moe():
4545
moe_block = model.model.language_model.layers[0].mlp
4646
assert isinstance(moe_block, LinearQwen3_5MoeSparseMoeBlock)
4747

48-
materialize_model_(model.model.language_model.layers[0])
48+
materialize_model(model.model.language_model.layers[0])
4949

5050
torch.testing.assert_close(moe_block.experts[0].gate_proj.weight, expected_gate)
5151
torch.testing.assert_close(moe_block.experts[0].up_proj.weight, expected_up)

0 commit comments

Comments
 (0)