Commit 99f9243d authored by Julien Chaumond's avatar Julien Chaumond
Browse files

same here, try to not serialize too much if unneeded

parent 9d8fd2d4
...@@ -503,7 +503,8 @@ class PreTrainedTokenizer(object): ...@@ -503,7 +503,8 @@ class PreTrainedTokenizer(object):
tokenizer_config_file = os.path.join(save_directory, TOKENIZER_CONFIG_FILE) tokenizer_config_file = os.path.join(save_directory, TOKENIZER_CONFIG_FILE)
tokenizer_config = copy.deepcopy(self.init_kwargs) tokenizer_config = copy.deepcopy(self.init_kwargs)
tokenizer_config["init_inputs"] = copy.deepcopy(self.init_inputs) if len(self.init_inputs) > 0:
tokenizer_config["init_inputs"] = copy.deepcopy(self.init_inputs)
for file_id in self.vocab_files_names.keys(): for file_id in self.vocab_files_names.keys():
tokenizer_config.pop(file_id, None) tokenizer_config.pop(file_id, None)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment