1 parent c60adf4 commit 98bfefeCopy full SHA for 98bfefe
1 file changed
fastdeploy/config.py
@@ -156,9 +156,7 @@ def __init__(
156
if hasattr(self, "vision_config"):
157
self.vision_config = PretrainedConfig.from_dict(self.vision_config)
158
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)
+ self.ori_vocab_size = args.get("ori_vocab_size", self.vocab_size)
162
163
architectures = self.architectures[0]
164
if MultimodalRegistry.contains_model(architectures):
0 commit comments