diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py index 1e53f35e8..5cbe6e885 100755 --- a/convert-hf-to-gguf.py +++ b/convert-hf-to-gguf.py @@ -1470,7 +1470,7 @@ class GrokModel(Model): @Model.register("DbrxForCausalLM") -class Qwen2MoeModel(Model): +class DbrxModel(Model): model_arch = gguf.MODEL_ARCH.DBRX def set_gguf_parameters(self):