integrations.expert_parallel.args

integrations.expert_parallel.args

Pydantic args for the Expert-Parallel (DeepEP) plugin.

Classes

Name Description
ExpertParallelArgs Input args for the expert_parallel plugin. See the integration README.

ExpertParallelArgs

integrations.expert_parallel.args.ExpertParallelArgs()

Input args for the expert_parallel plugin. See the integration README.

Attributes

Name Description
expert_parallel_size Number of EP ranks. 1 = disabled (default), > 1 = enabled.
expert_parallel_token_capacity Max tokens routed to any single expert per forward; the lowest-weight excess (token,expert)