zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-08-29 05:40:26 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Change the last_segment logic in tokenizer utils

Browse Source
This commit is contained in:
Angelos Katharopoulos 2024-12-12 10:37:29 -08:00
parent b444acfd69
commit 6ef5ca4ce5
1 changed files with 3 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
llms/mlx_lm/tokenizer_utils.py
View File

@ -3,8 +3,6 @@ from functools import partial
from transformers import AutoTokenizer
REPLACEMENT_CHAR = "\ufffd"
class StreamingDetokenizer:
"""The streaming detokenizer interface so that we can detokenize one token at a time.
@ -51,11 +49,9 @@ class StreamingDetokenizer:
def last_segment(self):
"""Return the last segment of readable text since last time this property was accessed."""
text = self.text
if text and text[-1] != REPLACEMENT_CHAR:
segment = text[self.offset :]
self.offset = len(text)
return segment
return ""
segment = text[self.offset :]
self.offset = len(text)
return text
class NaiveStreamingDetokenizer(StreamingDetokenizer):