def get_auto_awq_qlinear_cls(quantization_config):
    """Get the right AutoAWQQuantLinear class based on the quantization config."""
    if transformers.utils.import_utils.is_auto_awq_available():
        from awq.modules.linear import WQLinear_GEMM

        return WQLinear_GEMM
    return None
Get the right AutoAWQQuantLinear class based on the quantization config.
olive/common/hf/quant.py