From 9e6c61c647aaa6b234e2d6ae901caeddc0708d57 Mon Sep 17 00:00:00 2001
From: ckl117 <ckl117@163.com>
Date: Fri, 5 Sep 2025 13:07:08 +0800
Subject: [PATCH 1/2] fix

---
 fastdeploy/config.py                             |  1 +
 fastdeploy/model_executor/models/ernie4_5_moe.py | 13 +++++++++++++
 2 files changed, 14 insertions(+)

diff --git a/fastdeploy/config.py b/fastdeploy/config.py
index 2992312742a..7ad2052661c 100644
--- a/fastdeploy/config.py
+++ b/fastdeploy/config.py
@@ -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",
diff --git a/fastdeploy/model_executor/models/ernie4_5_moe.py b/fastdeploy/model_executor/models/ernie4_5_moe.py
index dcfd02c3e08..e9a07f2e9cd 100644
--- a/fastdeploy/model_executor/models/ernie4_5_moe.py
+++ b/fastdeploy/model_executor/models/ernie4_5_moe.py
@@ -645,6 +645,19 @@ def name(self):
         return "Ernie4_5_ForCausalLM"
 
 
+class Ernie4_5ForCausalLM(Ernie4_5_MoeForCausalLM):
+    """
+    Ernie4_5ForCausalLM 0.3B-PT
+    """
+
+    @classmethod
+    def name(self):
+        """
+        Model Architecture Name
+        """
+        return "Ernie4_5ForCausalLM"
+
+
 class Ernie4_5_MoePretrainedModel(PretrainedModel):
     """
     Ernie4_5_MoePretrainedModel

From d2da519ea5b1bb0d27cc947ceb8dffdbfd42a5d6 Mon Sep 17 00:00:00 2001
From: ckl117 <ckl117@163.com>
Date: Fri, 5 Sep 2025 14:47:44 +0800
Subject: [PATCH 2/2] check

---
 fastdeploy/model_executor/models/ernie4_5_moe.py | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/fastdeploy/model_executor/models/ernie4_5_moe.py b/fastdeploy/model_executor/models/ernie4_5_moe.py
index e9a07f2e9cd..bae92ea8fd7 100644
--- a/fastdeploy/model_executor/models/ernie4_5_moe.py
+++ b/fastdeploy/model_executor/models/ernie4_5_moe.py
@@ -645,7 +645,7 @@ def name(self):
         return "Ernie4_5_ForCausalLM"
 
 
-class Ernie4_5ForCausalLM(Ernie4_5_MoeForCausalLM):
+class Ernie4_5ForCausalLM(Ernie4_5_ForCausalLM):
     """
     Ernie4_5ForCausalLM 0.3B-PT
     """
@@ -809,3 +809,16 @@ def arch_name(self):
         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"