[Misc] Fix PhiMoE expert mapping (vllm-project#21085)

Signed-off-by: Jee Jee Li <[email protected]>
Signed-off-by: Jinzhen Lin <[email protected]>

Author: jeejeelee

vllm/model_executor/models/phimoe.py

@@ -533,14 +533,9 @@ def load_weights(self, weights: Iterable[tuple[str,
             ("qkv_proj", "v_proj", "v"),
         ]
 
-        expert_params_mapping = FusedMoE.make_expert_params_mapping(
-            ckpt_gate_proj_name="w1",
-            ckpt_down_proj_name="w2",
-            ckpt_up_proj_name="w3",
-            num_experts=self.config.num_local_experts)
-
         params_dict = dict(self.named_parameters())
         loaded_params: set[str] = set()
+        expert_params_mapping = self.get_expert_mapping()
         for name, loaded_weight in weights:
             if (self.quant_config is not None and
                 (scale_name := self.quant_config.get_cache_scale(name))):