From db2425a00b03eae56535328820352bf0e90dd4ed Mon Sep 17 00:00:00 2001 From: Julian Huang Date: Tue, 20 Jan 2026 20:08:28 +0800 Subject: [PATCH] [Fix]: correctly fetch ds32 config in tuning_fused_moe_triton (#17409) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: 墨楼 --- benchmark/kernels/fused_moe_triton/common_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/benchmark/kernels/fused_moe_triton/common_utils.py b/benchmark/kernels/fused_moe_triton/common_utils.py index b90d105c4..5f2d9aa8a 100644 --- a/benchmark/kernels/fused_moe_triton/common_utils.py +++ b/benchmark/kernels/fused_moe_triton/common_utils.py @@ -2,13 +2,13 @@ import json from typing import Dict, List, TypedDict import torch -from transformers import AutoConfig from sglang.srt.layers.moe.fused_moe_triton.fused_moe import get_config_dtype_str from sglang.srt.layers.moe.fused_moe_triton.fused_moe_triton_config import ( get_config_file_name, ) from sglang.srt.utils import is_hip +from sglang.srt.utils.hf_transformers_utils import get_config class BenchmarkConfig(TypedDict): @@ -36,7 +36,7 @@ def get_model_config( disable_shared_experts_fusion: bool = False, topk_ids_dir: str = None, ) -> Dict: - config = AutoConfig.from_pretrained(model_name, trust_remote_code=True) + config = get_config(model_name, trust_remote_code=True) block_shape = None if (