from vllm.model_executor.layers.quantization import QUANTIZATION_METHODS
from vllm_mlu.model_executor.layers.quantization.weightonly import WeightOnlyConfig
from vllm_mlu.model_executor.layers.quantization.smoothquant import SmoothQuantConfig
from vllm_mlu.model_executor.layers.quantization.gptq_mlu import GPTQMluConfig
from vllm_mlu.model_executor.layers.quantization.awq_mlu import AWQMluConfig


QUANTIZATION_METHODS.update({
    "gptq_mlu": GPTQMluConfig,
    "awq_mlu": AWQMluConfig,
    "weightonly": WeightOnlyConfig,
    "smoothquant": SmoothQuantConfig,
})
