Compatible with EB 0.3B torch model arch (#3913)

* fix

* check
This commit is contained in:
chen
2025-09-05 19:04:59 +08:00
committed by GitHub
parent bd7d15f7ea
commit 0d989829bb
2 changed files with 27 additions and 0 deletions

View File

@@ -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",