RangiLyu commited on
Commit
7b7548a
·
verified ·
1 Parent(s): 8882cdb

fix out of vocab token (#12)

Browse files

- fix out of vocab token (59045e3b2ef380d9abed3f44ba6349dc595f3550)

Files changed (1) hide show
  1. tokenization_interns1.py +4 -0
tokenization_interns1.py CHANGED
@@ -893,6 +893,10 @@ class InternS1Tokenizer(Qwen2Tokenizer):
893
  else:
894
  return self.encoder.get(token, self.encoder.get(self._unk_token))
895
 
 
 
 
 
896
  def convert_tokens_to_string(self, tokens):
897
  """Converts a sequence of tokens (string) in a single string."""
898
  text = "".join(tokens)
 
893
  else:
894
  return self.encoder.get(token, self.encoder.get(self._unk_token))
895
 
896
+ def _convert_id_to_token(self, index):
897
+ """Converts an index (integer) in a token (str) using the vocab."""
898
+ return self.decoder.get(index, "")
899
+
900
  def convert_tokens_to_string(self, tokens):
901
  """Converts a sequence of tokens (string) in a single string."""
902
  text = "".join(tokens)