Sfoglia il codice sorgente

convert : update Ernie 4.5 dense architecture name (#15555)

Signed-off-by: Weizhao Ouyang <o451686892@gmail.com>
Weizhao Ouyang 4 mesi fa
parent
commit
0d5a470223
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  1. 1 1
      convert_hf_to_gguf.py

+ 1 - 1
convert_hf_to_gguf.py

@@ -3159,7 +3159,7 @@ class LLaDAModel(TextModel):
         yield from super().modify_tensors(data_torch, name, bid)
 
 
-@ModelBase.register("Ernie4_5_ForCausalLM")
+@ModelBase.register("Ernie4_5_ForCausalLM", "Ernie4_5ForCausalLM")
 class Ernie4_5Model(TextModel):
     model_arch = gguf.MODEL_ARCH.ERNIE4_5