From 1fb6d95c1de98001f60ad932b58b8cb22a959019 Mon Sep 17 00:00:00 2001 From: Pierrick HYMBERT Date: Sun, 7 Apr 2024 15:40:21 +0200 Subject: [PATCH] model: convert-hf-to-gguf.py fix classname conflict with qwen2 --- convert-hf-to-gguf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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):