Refactored code

2026-08-26 15:16:23 +02:00 · 2024-04-13 19:33:06 +06:00
parent c8e7d9521d
commit 4c3e882a85
2 changed files with 2 additions and 0 deletions
@@ -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"]