From 8da487d5d03db6e01dd25fa3ae52608026ae815c Mon Sep 17 00:00:00 2001 From: Vikash Vardhan Date: Fri, 10 Jul 2026 11:03:55 -0500 Subject: [PATCH] fix: add BitNetForCausalLM alias and vocab fallback for BitnetModel --- utils/convert-hf-to-gguf-bitnet.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/utils/convert-hf-to-gguf-bitnet.py b/utils/convert-hf-to-gguf-bitnet.py index 23e84384c..a631f2b5d 100644 --- a/utils/convert-hf-to-gguf-bitnet.py +++ b/utils/convert-hf-to-gguf-bitnet.py @@ -952,13 +952,19 @@ def prepare_tensors(self): raise ValueError(f"Unprocessed experts: {experts}") -@Model.register("BitnetForCausalLM") +@Model.register("BitnetForCausalLM", "BitNetForCausalLM") class BitnetModel(Model): model_arch = gguf.MODEL_ARCH.BITNET def set_vocab(self): - self._set_vocab_sentencepiece() - + try: + self._set_vocab_sentencepiece() + except FileNotFoundError: + try: + self._set_vocab_llama_hf() + except (FileNotFoundError, TypeError): + # Llama 3 + self._set_vocab_gpt2() def set_gguf_parameters(self): super().set_gguf_parameters()