From 35b8362804234af8d7bee49a04fa9d6b62c5a9ac Mon Sep 17 00:00:00 2001 From: Zero Rains Date: Tue, 9 Sep 2025 15:07:51 +0800 Subject: [PATCH] get org_vocab_size from args (#3984) --- fastdeploy/config.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/fastdeploy/config.py b/fastdeploy/config.py index a0121afdf..f1ee1d457 100644 --- a/fastdeploy/config.py +++ b/fastdeploy/config.py @@ -155,9 +155,7 @@ class ModelConfig: if hasattr(self, "vision_config"): self.vision_config = PretrainedConfig.from_dict(self.vision_config) - self.ori_vocab_size = self.vocab_size - if ErnieArchitectures.contains_ernie_arch(self.architectures): - self.ori_vocab_size = args.get("ori_vocab_size", self.ori_vocab_size) + self.ori_vocab_size = args.get("ori_vocab_size", self.vocab_size) architectures = self.architectures[0] if MultimodalRegistry.contains_model(architectures):