From bd30b08521503a1e0df4389fe74f5734d60a0034 Mon Sep 17 00:00:00 2001 From: Zero Rains Date: Tue, 9 Sep 2025 15:08:47 +0800 Subject: [PATCH] get org_vocab_size from args (#3981) --- fastdeploy/config.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/fastdeploy/config.py b/fastdeploy/config.py index faf093b33..1f6b9b0ab 100644 --- a/fastdeploy/config.py +++ b/fastdeploy/config.py @@ -133,9 +133,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) class ParallelConfig: