瀏覽代碼

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

Co-authored-by: Csaba Kecskemeti <csabakecskemeti@Csabas-Mac-Pro.local>
Csaba Kecskemeti 1 年之前
父節點
當前提交
3c7989fd29
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      convert_hf_to_gguf.py

+ 1 - 1
convert_hf_to_gguf.py

@@ -1487,7 +1487,7 @@ class StableLMModel(Model):
                 raise ValueError(f"Unprocessed norms: {norms}")
                 raise ValueError(f"Unprocessed norms: {norms}")
 
 
 
 
-@Model.register("LlamaForCausalLM", "MistralForCausalLM", "MixtralForCausalLM")
+@Model.register("LLaMAForCausalLM", "LlamaForCausalLM", "MistralForCausalLM", "MixtralForCausalLM")
 class LlamaModel(Model):
 class LlamaModel(Model):
     model_arch = gguf.MODEL_ARCH.LLAMA
     model_arch = gguf.MODEL_ARCH.LLAMA