From: Csaba Kecskemeti <redacted>
Date: Sun, 15 Sep 2024 07:48:25 +0000 (-0700)
Subject: py : add "LLaMAForCausalLM" conversion support (#9485)
X-Git-Tag: upstream/0.0.4488~729
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=3c7989fd29a2db2b75e28fd708cc441febe99a82;p=pkg%2Fggml%2Fsources%2Fllama.cpp

py : add "LLaMAForCausalLM" conversion support (#9485)

Co-authored-by: Csaba Kecskemeti <redacted>
---

diff --git a/convert_hf_to_gguf.py b/convert_hf_to_gguf.py
index 01a8a50a..2c6d5d95 100755
--- a/convert_hf_to_gguf.py
+++ b/convert_hf_to_gguf.py
@@ -1487,7 +1487,7 @@ class StableLMModel(Model):
                 raise ValueError(f"Unprocessed norms: {norms}")
 
 
-@Model.register("LlamaForCausalLM", "MistralForCausalLM", "MixtralForCausalLM")
+@Model.register("LLaMAForCausalLM", "LlamaForCausalLM", "MistralForCausalLM", "MixtralForCausalLM")
 class LlamaModel(Model):
     model_arch = gguf.MODEL_ARCH.LLAMA