[BugFix] fix video bug (#5557)

* fix video bug

* add eb5 moe model
This commit is contained in:
kevin
2025-12-16 20:06:50 +08:00
committed by GitHub
parent 27ef3610b5
commit 7140939c51
2 changed files with 5 additions and 3 deletions

View File

@@ -245,7 +245,7 @@ class ModelConfig:
"""
check if the model architecture disable for mm prefill
"""
return self._architecture in ["Ernie5ForCausalLM"]
return self._architecture in ["Ernie5ForCausalLM", "Ernie5MoeForCausalLM"]
def _post_init(self):
self.is_unified_ckpt = check_unified_ckpt(self.model)