use self.hidden_size not use self.fd_config.model_config.hidden_size (#7340)

2026-04-23 00:17:25 +08:00 · 2026-04-11 22:39:43 +08:00
parent 4982aa000e
commit 225fc8d222
2 changed files with 3 additions and 1 deletions
@@ -169,6 +169,8 @@ class TestChunkedMoE(unittest.TestCase):
        fused_moe.fd_config = mock_fd_config
        fused_moe.quant_method = MockQuantMethod()
        fused_moe.enable_routing_replay = None
+
+        fused_moe.hidden_size = mock_fd_config.model_config.hidden_size
        return fused_moe

    def run_model_runner(self):