add import warnings

hpcaitech · Aug 17, 2023 · cdb3181 · cdb3181
1 parent 6ce0b81
commit cdb3181
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/colossalai/gptq/cai_gptq/__init__.py b/colossalai/gptq/cai_gptq/__init__.py
@@ -1,15 +1,14 @@
+import warnings
 
 HAS_AUTO_GPTQ = False
-
 try:
     import auto_gptq
     HAS_AUTO_GPTQ = True
 except ImportError:
     warnings.warn('please install auto-gptq from https://github.com/PanQiWei/AutoGPTQ')
     HAS_AUTO_GPTQ = False
-
-if HAS_AUTO_GPTQ: 
 
+if HAS_AUTO_GPTQ:
     from .gptq_triton import gptq_fused_linear_triton
     from .cai_quant_linear import make_cai_quant_linear, CaiQuantLinear
     from .gptq_op import CaiGPTQLinearOp