Fix quantized moe checker fail for Qwen3 dense fp8 model (#13853)
This commit is contained in:
@@ -597,9 +597,12 @@ class ModelRunner:
|
|||||||
)
|
)
|
||||||
moe_tp_size = self.tp_size // self.moe_ep_size
|
moe_tp_size = self.tp_size // self.moe_ep_size
|
||||||
|
|
||||||
moe_intermediate_size = (
|
moe_intermediate_size = getattr(
|
||||||
self.model_config.hf_text_config.moe_intermediate_size
|
self.model_config.hf_text_config, "moe_intermediate_size", None
|
||||||
)
|
)
|
||||||
|
if moe_intermediate_size is None:
|
||||||
|
return
|
||||||
|
|
||||||
if moe_intermediate_size % moe_tp_size != 0:
|
if moe_intermediate_size % moe_tp_size != 0:
|
||||||
raise ValueError(
|
raise ValueError(
|
||||||
f"moe_intermediate_size {moe_intermediate_size} must be divisible by moe_tp_size ({moe_tp_size}) which is tp_size ({self.tp_size}) divided by moe_ep_size ({self.moe_ep_size})."
|
f"moe_intermediate_size {moe_intermediate_size} must be divisible by moe_tp_size ({moe_tp_size}) which is tp_size ({self.tp_size}) divided by moe_ep_size ({self.moe_ep_size})."
|
||||||
|
|||||||
Reference in New Issue
Block a user