diff --git a/utils/convert-hf-to-gguf-bitnet.py b/utils/convert-hf-to-gguf-bitnet.py index 23e84384c..f1c9d1720 100644 --- a/utils/convert-hf-to-gguf-bitnet.py +++ b/utils/convert-hf-to-gguf-bitnet.py @@ -952,7 +952,7 @@ def prepare_tensors(self): raise ValueError(f"Unprocessed experts: {experts}") -@Model.register("BitnetForCausalLM") +@Model.register("BitNetForCausalLM") class BitnetModel(Model): model_arch = gguf.MODEL_ARCH.BITNET diff --git a/utils/generate-dummy-bitnet-model.py b/utils/generate-dummy-bitnet-model.py index be3f6cdaa..203d9bbb8 100644 --- a/utils/generate-dummy-bitnet-model.py +++ b/utils/generate-dummy-bitnet-model.py @@ -773,7 +773,7 @@ def preprocess_weights_tl2( return weight -@Model.register("BitnetForCausalLM") +@Model.register("BitNetForCausalLM") class BitnetModel(Model): model_arch = gguf.MODEL_ARCH.BITNET params: str = ""