Refactored code

2024-04-13 19:33:06 +06:00 · 2024-04-13 19:33:06 +06:00 · e5dda78bf5
commit e5dda78bf5
parent 360849169f
2 changed files with 2 additions and 0 deletions
--- a/convert-hf-to-gguf.py
+++ b/convert-hf-to-gguf.py
@ -1303,6 +1303,8 @@ class PersimmonModel(Model):
            self.gguf_writer.add_tensor(new_name, data)
@Model.register("LlamaForCausalLM")
 class DeepseekCoderModel(Model):
+    model_arch = gguf.MODEL_ARCH.LLAMA
+    
    def set_gguf_parameters(self):
        super().set_gguf_parameters()
        head_count = self.hparams["num_attention_heads"]
--- a/models/ggml-vocab-deepseek-coder.gguf
+++ b/models/ggml-vocab-deepseek-coder.gguf