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

* fix * check
2025-09-27 04:46:16 +08:00 · 2025-09-05 19:05:07 +08:00
parent 173e4df982
commit 335d1c8e8f
2 changed files with 27 additions and 0 deletions
--- 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",