Refactored code

2024-04-13 19:33:06 +06:00 · 2024-04-13 19:33:06 +06:00 · 4c3e882a85
commit 4c3e882a85
parent c8e7d9521d
2 changed files with 2 additions and 0 deletions
--- a/convert-hf-to-gguf.py
+++ b/convert-hf-to-gguf.py
@ -1314,6 +1314,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