mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-05 00:33:03 +08:00
@@ -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",
|
||||
|
@@ -645,6 +645,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
|
||||
@@ -796,3 +809,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"
|
||||
|
Reference in New Issue
Block a user