simcop2387/AutoGPTQ
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity

improve code format

Browse source
This commit is contained in:
PanQiWei 2023-08-13 16:14:15 +08:00
parent 0a04d3fb2a
commit d95661b250
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
auto_gptq/modeling/gpt_bigcode.py
View file

@ -1,4 +1,4 @@
from auto_gptq.modeling import BaseGPTQForCausalLM from ._base import BaseGPTQForCausalLM
class GPTBigCodeGPTQForCausalLM(BaseGPTQForCausalLM): class GPTBigCodeGPTQForCausalLM(BaseGPTQForCausalLM):
@ -14,4 +14,5 @@ class GPTBigCodeGPTQForCausalLM(BaseGPTQForCausalLM):
["mlp.c_proj"] ["mlp.c_proj"]
] ]
__all__ = ["GPTBigCodeGPTQForCausalLM"] __all__ = ["GPTBigCodeGPTQForCausalLM"]