Fix EOS override persistence on save and reload

#2
by edbeeching HF Staff - opened
Files changed (1) hide show
  1. tokenizer.py +4 -0
tokenizer.py CHANGED
@@ -716,6 +716,7 @@ class HybridDNATokenizer(PreTrainedTokenizer):
716
 
717
  # Save base tokenizer files with the wrapper's effective chat template.
718
  self._base_tokenizer.chat_template = self.chat_template
 
719
  self._base_tokenizer.save_pretrained(save_directory)
720
 
721
  # Save DNA config
@@ -816,6 +817,9 @@ class HybridDNATokenizer(PreTrainedTokenizer):
816
  tokenizer_config = json.load(_f)
817
  if tokenizer_config.get("chat_template"):
818
  _init["chat_template"] = tokenizer_config["chat_template"]
 
 
 
819
 
820
  chat_template_path = cached_file(pretrained_model_name_or_path, "chat_template.jinja", **hub_kwargs)
821
  if chat_template_path is not None:
 
716
 
717
  # Save base tokenizer files with the wrapper's effective chat template.
718
  self._base_tokenizer.chat_template = self.chat_template
719
+ self._base_tokenizer.eos_token = self.eos_token
720
  self._base_tokenizer.save_pretrained(save_directory)
721
 
722
  # Save DNA config
 
817
  tokenizer_config = json.load(_f)
818
  if tokenizer_config.get("chat_template"):
819
  _init["chat_template"] = tokenizer_config["chat_template"]
820
+ if tokenizer_config.get("eos_token") is not None:
821
+ eos_token = tokenizer_config["eos_token"]
822
+ _init["eos_token"] = eos_token["content"] if isinstance(eos_token, dict) else eos_token
823
 
824
  chat_template_path = cached_file(pretrained_model_name_or_path, "chat_template.jinja", **hub_kwargs)
825
  if chat_template_path is not None: