Skip to content

Commit 335d1c8

Browse files
authored
【CP】Compatible with EB 0.3B torch model arch (#3914)
* fix * check
1 parent 173e4df commit 335d1c8

File tree

2 files changed

+27
-0
lines changed

2 files changed

+27
-0
lines changed

fastdeploy/config.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,7 @@ class ErnieArchitectures:
6262
"""Helper class for ERNIE architecture check."""
6363

6464
ARCHITECTURES = {
65+
"Ernie4_5ForCausalLM", # 0.3B-PT
6566
"Ernie4_5_ForCausalLM",
6667
"Ernie4_5_MoeForCausalLM",
6768
"Ernie4_5_VLMoeForConditionalGeneration",

fastdeploy/model_executor/models/ernie4_5_moe.py

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -637,6 +637,19 @@ def name(self):
637637
return "Ernie4_5_ForCausalLM"
638638

639639

640+
class Ernie4_5ForCausalLM(Ernie4_5_ForCausalLM):
641+
"""
642+
Ernie4_5ForCausalLM 0.3B-PT
643+
"""
644+
645+
@classmethod
646+
def name(self):
647+
"""
648+
Model Architecture Name
649+
"""
650+
return "Ernie4_5ForCausalLM"
651+
652+
640653
class Ernie4_5_MoePretrainedModel(PretrainedModel):
641654
"""
642655
Ernie4_5_MoePretrainedModel
@@ -788,3 +801,16 @@ def arch_name(self):
788801
Model Architecture Name
789802
"""
790803
return "Ernie4_5_ForCausalLM"
804+
805+
806+
class Ernie4_5PretrainedModel(Ernie4_5_PretrainedModel):
807+
"""
808+
Ernie4_5PretrainedModel 0.3B-PT
809+
"""
810+
811+
@classmethod
812+
def arch_name(self):
813+
"""
814+
Model Architecture Name
815+
"""
816+
return "Ernie4_5ForCausalLM"

0 commit comments

Comments
 (0)