github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-28 21:17:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
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
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: