diff --git a/hanlp/common/torch_component.py b/hanlp/common/torch_component.py index 7750ab1d7..82ccf1be0 100644 --- a/hanlp/common/torch_component.py +++ b/hanlp/common/torch_component.py @@ -248,9 +248,9 @@ def fit(self, _dummy_placeholder = self._create_dummy_placeholder_on(first_device) if finetune: if isinstance(finetune, str): - self.load(finetune, devices=devices) + self.load(finetune, devices=devices, **self.config) else: - self.load(save_dir, devices=devices) + self.load(save_dir, devices=devices, **self.config) self.config.finetune = finetune self.vocabs.unlock() # For extending vocabs logger.info(