fix arguement error in ep when pd (#3060)

This commit is contained in:
Longzhi Wang
2025-07-29 17:17:24 +08:00
committed by GitHub
parent 502ee92a0a
commit be0a0f2bb2

View File

@@ -66,7 +66,7 @@ class MoEMethodBase(QuantMethodBase):
layer.fd_config.model_config.redundant_experts_num,
)
else:
if layer.fd_config.parallel_config.moe_phase == "prefill":
if layer.fd_config.parallel_config.moe_phase.phase == "prefill":
from .ep import EPPrefillRunner
self.ep_prefill_runner = EPPrefillRunner(