Pārlūkot izejas kodu

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

Signed-off-by: Weizhao Ouyang <o451686892@gmail.com>
Weizhao Ouyang 4 mēneši atpakaļ
vecāks
revīzija
0d5a470223
1 mainītis faili ar 1 papildinājumiem un 1 dzēšanām
  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)
         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):
 class Ernie4_5Model(TextModel):
     model_arch = gguf.MODEL_ARCH.ERNIE4_5
     model_arch = gguf.MODEL_ARCH.ERNIE4_5