vllm_omni.diffusion.models.hunyuan_image3.hunyuan_fused_moe ¶

make_expert_params_mapping(
    model: Any,
    ckpt_gate_proj_name: str,
    ckpt_down_proj_name: str,
    ckpt_up_proj_name: str,
    num_experts: int,
    num_redundant_experts: int = 0,
) -> list[tuple[str, str, int, str]]

forward(hidden_states: Any, router_logits: Any) -> Any