File tree Expand file tree Collapse file tree 2 files changed +27
-0
lines changed Expand file tree Collapse file tree 2 files changed +27
-0
lines changed Original file line number Diff line number Diff line change @@ -62,6 +62,7 @@ class ErnieArchitectures:
62
62
"""Helper class for ERNIE architecture check."""
63
63
64
64
ARCHITECTURES = {
65
+ "Ernie4_5ForCausalLM" , # 0.3B-PT
65
66
"Ernie4_5_ForCausalLM" ,
66
67
"Ernie4_5_MoeForCausalLM" ,
67
68
"Ernie4_5_VLMoeForConditionalGeneration" ,
Original file line number Diff line number Diff line change @@ -637,6 +637,19 @@ def name(self):
637
637
return "Ernie4_5_ForCausalLM"
638
638
639
639
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
+
640
653
class Ernie4_5_MoePretrainedModel (PretrainedModel ):
641
654
"""
642
655
Ernie4_5_MoePretrainedModel
@@ -788,3 +801,16 @@ def arch_name(self):
788
801
Model Architecture Name
789
802
"""
790
803
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"
You can’t perform that action at this time.
0 commit comments