From d95661b25048bbd85b6b967a695fde2c6539524c Mon Sep 17 00:00:00 2001 From: PanQiWei <594557445@qq.com> Date: Sun, 13 Aug 2023 16:14:15 +0800 Subject: [PATCH] improve code format --- auto_gptq/modeling/gpt_bigcode.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/auto_gptq/modeling/gpt_bigcode.py b/auto_gptq/modeling/gpt_bigcode.py index f9afee6..19a1239 100644 --- a/auto_gptq/modeling/gpt_bigcode.py +++ b/auto_gptq/modeling/gpt_bigcode.py @@ -1,4 +1,4 @@ -from auto_gptq.modeling import BaseGPTQForCausalLM +from ._base import BaseGPTQForCausalLM class GPTBigCodeGPTQForCausalLM(BaseGPTQForCausalLM): @@ -14,4 +14,5 @@ class GPTBigCodeGPTQForCausalLM(BaseGPTQForCausalLM): ["mlp.c_proj"] ] -__all__ = ["GPTBigCodeGPTQForCausalLM"] \ No newline at end of file + +__all__ = ["GPTBigCodeGPTQForCausalLM"]