【CP】Compatible with EB 0.3B torch model arch (#3914)

* fix

* check
This commit is contained in:
chen
2025-09-05 19:05:07 +08:00
committed by GitHub
parent 173e4df982
commit 335d1c8e8f
2 changed files with 27 additions and 0 deletions

View File

@@ -62,6 +62,7 @@ class ErnieArchitectures:
"""Helper class for ERNIE architecture check."""
ARCHITECTURES = {
"Ernie4_5ForCausalLM", # 0.3B-PT
"Ernie4_5_ForCausalLM",
"Ernie4_5_MoeForCausalLM",
"Ernie4_5_VLMoeForConditionalGeneration",

View File

@@ -637,6 +637,19 @@ class Ernie4_5_ForCausalLM(Ernie4_5_MoeForCausalLM):
return "Ernie4_5_ForCausalLM"
class Ernie4_5ForCausalLM(Ernie4_5_ForCausalLM):
"""
Ernie4_5ForCausalLM 0.3B-PT
"""
@classmethod
def name(self):
"""
Model Architecture Name
"""
return "Ernie4_5ForCausalLM"
class Ernie4_5_MoePretrainedModel(PretrainedModel):
"""
Ernie4_5_MoePretrainedModel
@@ -788,3 +801,16 @@ class Ernie4_5_PretrainedModel(Ernie4_5_MoePretrainedModel):
Model Architecture Name
"""
return "Ernie4_5_ForCausalLM"
class Ernie4_5PretrainedModel(Ernie4_5_PretrainedModel):
"""
Ernie4_5PretrainedModel 0.3B-PT
"""
@classmethod
def arch_name(self):
"""
Model Architecture Name
"""
return "Ernie4_5ForCausalLM"