Compare commits
1
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
49a43e0c7b |
@@ -1352,6 +1352,15 @@ class PreTrainedTokenizerBase(SpecialTokensMixin):
|
|||||||
added_tokens_file = os.path.join(save_directory, ADDED_TOKENS_FILE)
|
added_tokens_file = os.path.join(save_directory, ADDED_TOKENS_FILE)
|
||||||
tokenizer_config_file = os.path.join(save_directory, TOKENIZER_CONFIG_FILE)
|
tokenizer_config_file = os.path.join(save_directory, TOKENIZER_CONFIG_FILE)
|
||||||
|
|
||||||
|
def convert_values(dic):
|
||||||
|
write_dict = {}
|
||||||
|
for key, value in dic.items():
|
||||||
|
if isinstance(value, AddedToken):
|
||||||
|
write_dict[key] = value.__getstate__()
|
||||||
|
else:
|
||||||
|
write_dict[key] = value
|
||||||
|
return write_dict
|
||||||
|
|
||||||
tokenizer_config = copy.deepcopy(self.init_kwargs)
|
tokenizer_config = copy.deepcopy(self.init_kwargs)
|
||||||
if len(self.init_inputs) > 0:
|
if len(self.init_inputs) > 0:
|
||||||
tokenizer_config["init_inputs"] = copy.deepcopy(self.init_inputs)
|
tokenizer_config["init_inputs"] = copy.deepcopy(self.init_inputs)
|
||||||
@@ -1359,16 +1368,10 @@ class PreTrainedTokenizerBase(SpecialTokensMixin):
|
|||||||
tokenizer_config.pop(file_id, None)
|
tokenizer_config.pop(file_id, None)
|
||||||
|
|
||||||
with open(tokenizer_config_file, "w", encoding="utf-8") as f:
|
with open(tokenizer_config_file, "w", encoding="utf-8") as f:
|
||||||
f.write(json.dumps(tokenizer_config, ensure_ascii=False))
|
f.write(json.dumps(convert_values(tokenizer_config), ensure_ascii=False))
|
||||||
|
|
||||||
with open(special_tokens_map_file, "w", encoding="utf-8") as f:
|
with open(special_tokens_map_file, "w", encoding="utf-8") as f:
|
||||||
write_dict = {}
|
f.write(json.dumps(convert_values(self.special_tokens_map_extended), ensure_ascii=False))
|
||||||
for key, value in self.special_tokens_map_extended.items():
|
|
||||||
if isinstance(value, AddedToken):
|
|
||||||
write_dict[key] = value.__getstate__()
|
|
||||||
else:
|
|
||||||
write_dict[key] = value
|
|
||||||
f.write(json.dumps(write_dict, ensure_ascii=False))
|
|
||||||
|
|
||||||
added_vocab = self.get_added_vocab()
|
added_vocab = self.get_added_vocab()
|
||||||
if added_vocab:
|
if added_vocab:
|
||||||
|
|||||||
Reference in New Issue
Block a user