apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-05 16:48:03 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

get org_vocab_size from args (#3985)

Browse Source 
Co-authored-by: Yuanle Liu <yuanlehome@163.com>
This commit is contained in:
Zero Rains
2025-09-09 15:08:58 +08:00
committed by GitHub
parent 7d1b2bd732
commit d8985a7a21
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
fastdeploy/config.py
View File

@@ -155,9 +155,7 @@ class ModelConfig:
if hasattr(self, "vision_config"): if hasattr(self, "vision_config"):
self.vision_config = PretrainedConfig.from_dict(self.vision_config) self.vision_config = PretrainedConfig.from_dict(self.vision_config)
self.ori_vocab_size = self.vocab_size self.ori_vocab_size = args.get("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)
architectures = self.architectures[0] architectures = self.architectures[0]
if MultimodalRegistry.contains_model(architectures): if MultimodalRegistry.contains_model(architectures):