From 07ed155e8633e7aaaea83a0dee3da04119287a9a Mon Sep 17 00:00:00 2001 From: Michelle <97082656+MichelleMa8@users.noreply.github.com> Date: Wed, 27 Sep 2023 10:33:13 +0800 Subject: [PATCH] [NFC] polish colossalai/inference/quant/gptq/cai_gptq/__init__.py code style (#4792) --- colossalai/inference/quant/gptq/cai_gptq/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/colossalai/inference/quant/gptq/cai_gptq/__init__.py b/colossalai/inference/quant/gptq/cai_gptq/__init__.py index de57f2d8c..4ed76293b 100644 --- a/colossalai/inference/quant/gptq/cai_gptq/__init__.py +++ b/colossalai/inference/quant/gptq/cai_gptq/__init__.py @@ -3,9 +3,10 @@ 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') + warnings.warn("please install auto-gptq from https://github.com/PanQiWei/AutoGPTQ") HAS_AUTO_GPTQ = False if HAS_AUTO_GPTQ: