[Docs] Clarify Hunyuan expert remapping scope

Signed-off-by: hsliuustc0106 <liuhongsheng4@huawei.com>
This commit is contained in:
hsliuustc0106 committed 2026-09-03 23:02:46 +00:00
1 parent cecaa6344d
commit ade3bb27ea
3 files changed
+3 -1

No files matched your search

@@ -97,7 +97,7 @@ def test_model_loaders_do_not_call_removed_get_cache_scale(rel_path: str) -> Non
),
],
)
def test_hunyuan_expert_mapping_uses_vllm_public_contract(
def test_hunyuan_expert_mapping_separates_public_and_local_contracts(
module_name: str,
class_name: str,
monkeypatch: pytest.MonkeyPatch,
@@ -2205,6 +2205,7 @@ class HunyuanImage3Model(nn.Module):
)
def _get_expert_weights_remapping(self) -> dict[str, tuple[str, int, int]]:
"""Return the vLLM-Omni-local remapping for Hunyuan checkpoint weights."""
if not _is_moe(self.config):
return {}
return {
@@ -144,6 +144,7 @@ class HunyuanModel(HunYuanModel):
)
def _get_expert_weights_remapping(self) -> dict[str, tuple[str, int, int]]:
"""Return the vLLM-Omni-local remapping for Hunyuan checkpoint weights."""
if not _is_moe(self.config):
return {}
return {