Move import up

awq/models/base.py

@@ -8,6 +8,7 @@ from typing import List, Union
 from safetensors.torch import save_file
 from awq.modules.act import ScaledActivation
 from huggingface_hub import snapshot_download
+from awq.quantize.quantizer import AwqQuantizer
 from awq.utils.utils import simple_dispatch_model
 from transformers.modeling_utils import shard_checkpoint
 from awq.modules.linear import WQLinear_GEMM, WQLinear_GEMV
@@ -41,8 +42,6 @@ class BaseAWQForCausalLM(nn.Module):
         self.quant_config = quant_config
         quant_config["version"] = "GEMM" if 'version' not in quant_config.keys() else quant_config["version"]
 
-        from awq.quantize.quantizer import AwqQuantizer
-
         quantizer = AwqQuantizer(
             self, self.model, tokenizer, quant_config["w_bit"], quant_config["q_group_size"],
             quant_config["version"], calib_data, split, text_column