Skip to content

Commit 98bfefe

Browse files
authored
get org_vocab_size from args (PaddlePaddle#3983)
1 parent c60adf4 commit 98bfefe

1 file changed

Lines changed: 1 addition & 3 deletions

File tree

fastdeploy/config.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -156,9 +156,7 @@ def __init__(
156156
if hasattr(self, "vision_config"):
157157
self.vision_config = PretrainedConfig.from_dict(self.vision_config)
158158

159-
self.ori_vocab_size = self.vocab_size
160-
if ErnieArchitectures.contains_ernie_arch(self.architectures):
161-
self.ori_vocab_size = args.get("ori_vocab_size", self.ori_vocab_size)
159+
self.ori_vocab_size = args.get("ori_vocab_size", self.vocab_size)
162160

163161
architectures = self.architectures[0]
164162
if MultimodalRegistry.contains_model(architectures):

0 commit comments

Comments
 (0)