Skip to content

Commit 0d5a470

Browse files
authored
convert : update Ernie 4.5 dense architecture name (ggml-org#15555)
Signed-off-by: Weizhao Ouyang <[email protected]>
1 parent b0ba31f commit 0d5a470

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

convert_hf_to_gguf.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3159,7 +3159,7 @@ def modify_tensors(self, data_torch: Tensor, name: str, bid: int | None) -> Iter
31593159
yield from super().modify_tensors(data_torch, name, bid)
31603160

31613161

3162-
@ModelBase.register("Ernie4_5_ForCausalLM")
3162+
@ModelBase.register("Ernie4_5_ForCausalLM", "Ernie4_5ForCausalLM")
31633163
class Ernie4_5Model(TextModel):
31643164
model_arch = gguf.MODEL_ARCH.ERNIE4_5
31653165

0 commit comments

Comments
 (0)