get org_vocab_size from args (#3984)
Some checks failed
CE Compile Job / ce_job_pre_check (push) Has been cancelled
CE Compile Job / print_ce_job_pre_check_outputs (push) Has been cancelled
CE Compile Job / FD-Clone-Linux (push) Has been cancelled
CE Compile Job / Show Code Archive Output (push) Has been cancelled
CE Compile Job / BUILD_SM8090 (push) Has been cancelled
CE Compile Job / BUILD_SM8689 (push) Has been cancelled
CE Compile Job / CE_UPLOAD (push) Has been cancelled

This commit is contained in:
Zero Rains
2025-09-09 15:07:51 +08:00
committed by GitHub
parent d43c2f2577
commit 35b8362804

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):