diff --git a/vllm/models/qwen4_exp/common/mtp.py b/vllm/models/qwen4_exp/common/mtp.py index 457e8bc064..0aa22c4bdd 100644 --- a/vllm/models/qwen4_exp/common/mtp.py +++ b/vllm/models/qwen4_exp/common/mtp.py @@ -5,7 +5,7 @@ import torch from vllm.config import VllmConfig -from vllm.transformers_utils.configs.qwen4_exp import Qwen4ExpTextConfig +from transformers import Qwen4ExpTextConfig def make_mtp_hidden_buffer( diff --git a/vllm/models/qwen4_exp/nvidia/ple_pull.py b/vllm/models/qwen4_exp/nvidia/ple_pull.py index 25d9640b0c..1acc3084a8 100644 --- a/vllm/models/qwen4_exp/nvidia/ple_pull.py +++ b/vllm/models/qwen4_exp/nvidia/ple_pull.py @@ -35,7 +35,7 @@ from vllm.model_executor.layers.ple_offload_layer import ( is_offload_process, ) from vllm.model_executor.layers.quantization.base_config import QuantizationConfig -from vllm.transformers_utils.configs.qwen4_exp import Qwen4ExpTextConfig +from transformers import Qwen4ExpTextConfig from .ngram_embedding import ( Qwen4ExpNGramEmbedding, diff --git a/vllm/models/qwen4_exp/nvidia/qsa.py b/vllm/models/qwen4_exp/nvidia/qsa.py index f0d8979fc1..e4942c0ba2 100644 --- a/vllm/models/qwen4_exp/nvidia/qsa.py +++ b/vllm/models/qwen4_exp/nvidia/qsa.py @@ -40,9 +40,6 @@ from vllm.model_executor.models.qwen3_next import Qwen3NextAttention from vllm.model_executor.models.utils import AutoWeightsLoader, WeightsMapper from vllm.platforms import current_platform from vllm.platforms.interface import DeviceCapability -from vllm.transformers_utils.configs.qwen4_exp import ( - Qwen4ExpTextConfig, -) from vllm.utils.torch_utils import ( LayerNameType, _encode_layer_name,