github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[NFC] polish colossalai/inference/quant/gptq/cai_gptq/__init__.py code style (#4792)

pull/4887/head
Michelle 2023-09-27 10:33:13 +08:00 committed by Zian(Andy) Zheng
parent eef96e0877
commit 07ed155e86
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
colossalai/inference/quant/gptq/cai_gptq/__init__.py
View File

@ -3,9 +3,10 @@ import warnings
HAS_AUTO_GPTQ = False HAS_AUTO_GPTQ = False
try: try:
import auto_gptq import auto_gptq
HAS_AUTO_GPTQ = True HAS_AUTO_GPTQ = True
except ImportError: except ImportError:
warnings.warn('please install auto-gptq from https://github.com/PanQiWei/AutoGPTQ') warnings.warn("please install auto-gptq from https://github.com/PanQiWei/AutoGPTQ")
HAS_AUTO_GPTQ = False HAS_AUTO_GPTQ = False
if HAS_AUTO_GPTQ: if HAS_AUTO_GPTQ: