]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commitdiff
py : add "LLaMAForCausalLM" conversion support (#9485)
authorCsaba Kecskemeti <redacted>
Sun, 15 Sep 2024 07:48:25 +0000 (00:48 -0700)
committerGitHub <redacted>
Sun, 15 Sep 2024 07:48:25 +0000 (10:48 +0300)
Co-authored-by: Csaba Kecskemeti <redacted>
convert_hf_to_gguf.py

index 01a8a50a27cc630fed36a89866b395393bd0b777..2c6d5d95b8ba3a14bfe2970e02f97ae59985d11f 100755 (executable)
@@ -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