model: convert-hf-to-gguf.py fix classname conflict with qwen2

This commit is contained in:
Pierrick HYMBERT 2024-04-07 15:40:21 +02:00
parent 61be4b91a6
commit 1fb6d95c1d

View file

@ -1470,7 +1470,7 @@ class GrokModel(Model):
@Model.register("DbrxForCausalLM") @Model.register("DbrxForCausalLM")
class Qwen2MoeModel(Model): class DbrxModel(Model):
model_arch = gguf.MODEL_ARCH.DBRX model_arch = gguf.MODEL_ARCH.DBRX
def set_gguf_parameters(self): def set_gguf_parameters(self):