From 98bfefea021bfdb8000183b3bbd8f5ad83381462 Mon Sep 17 00:00:00 2001 From: Zero Rains Date: Tue, 9 Sep 2025 15:08:03 +0800 Subject: [PATCH] get org_vocab_size from args (#3983) --- fastdeploy/config.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/fastdeploy/config.py b/fastdeploy/config.py index 105242058..8400e90da 100644 --- a/fastdeploy/config.py +++ b/fastdeploy/config.py @@ -156,9 +156,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):