From 3f3c84fa6de40188d5c740d9d9fabf6f0c286a8f Mon Sep 17 00:00:00 2001 From: KexiongYu Date: Fri, 15 May 2026 15:13:31 +0800 Subject: [PATCH] Fix Hunyuan Image3 topk unpack compatibility Signed-off-by: KexiongYu --- .../model_executor/models/hunyuan_image3/hunyuan_image3.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm_omni/model_executor/models/hunyuan_image3/hunyuan_image3.py b/vllm_omni/model_executor/models/hunyuan_image3/hunyuan_image3.py index 0c2afb2ac34..a59b30e4427 100644 --- a/vllm_omni/model_executor/models/hunyuan_image3/hunyuan_image3.py +++ b/vllm_omni/model_executor/models/hunyuan_image3/hunyuan_image3.py @@ -1167,7 +1167,7 @@ def _hunyuan_image3_unpack_packed_topk( gating_output: torch.Tensor, topk: int, renormalize: bool, - num_experts: int, + num_experts: int | None = None, ) -> tuple[torch.Tensor, torch.Tensor]: """Unpack pre-computed ``(topk_weights, topk_indices)`` packed by :class:`HunyuanImage3SparseMoeBlock` into ``gating_output``.