fix out of vocab token (#12)
Browse files- fix out of vocab token (59045e3b2ef380d9abed3f44ba6349dc595f3550)
- 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)
|