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

Rename the class to match reference capitalisation

Browse source
This commit is contained in:
LaaZa 2023-06-18 21:01:07 +03:00
parent 9fd558f2ba
commit 03577a7698
2 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
auto_gptq/modeling/auto.py
View file

@ -13,7 +13,7 @@ from .moss import MOSSGPTQForCausalLM
from .opt import OPTGPTQForCausalLM from .opt import OPTGPTQForCausalLM
from .rw import RWGPTQForCausalLM from .rw import RWGPTQForCausalLM
from .gpt_bigcode import GPTBigCodeGPTQForCausalLM from .gpt_bigcode import GPTBigCodeGPTQForCausalLM
from .baichuan import BaichuanGPTQForCausalLM from .baichuan import BaiChuanGPTQForCausalLM
GPTQ_CAUSAL_LM_MODEL_MAP = { GPTQ_CAUSAL_LM_MODEL_MAP = {
@ -28,7 +28,7 @@ GPTQ_CAUSAL_LM_MODEL_MAP = {
"codegen": CodeGenGPTQForCausalLM, "codegen": CodeGenGPTQForCausalLM,
"RefinedWebModel": RWGPTQForCausalLM, "RefinedWebModel": RWGPTQForCausalLM,
"RefinedWeb":RWGPTQForCausalLM, "RefinedWeb":RWGPTQForCausalLM,
"baichuan":BaichuanGPTQForCausalLM "baichuan":BaiChuanGPTQForCausalLM
} }

4
auto_gptq/modeling/baichuan.py
View file

@ -1,7 +1,7 @@
from ._base import * from ._base import *
class BaichuanGPTQForCausalLM(BaseGPTQForCausalLM): class BaiChuanGPTQForCausalLM(BaseGPTQForCausalLM):
layer_type = "DecoderLayer" layer_type = "DecoderLayer"
layers_block_name = "model.layers" layers_block_name = "model.layers"
outside_layer_modules = ["model.embed_tokens", "model.norm"] outside_layer_modules = ["model.embed_tokens", "model.norm"]
@ -13,4 +13,4 @@ class BaichuanGPTQForCausalLM(BaseGPTQForCausalLM):
] ]
__all__ = ["BaichuanGPTQForCausalLM"] __all__ = ["BaiChuanGPTQForCausalLM"]