We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 991143c commit 203d4f8Copy full SHA for 203d4f8
vllm/transformers_utils/tokenizer.py
@@ -26,6 +26,7 @@ def get_cached_tokenizer(
26
tokenizer_all_special_tokens_extended = (
27
tokenizer.all_special_tokens_extended)
28
tokenizer_all_special_tokens = set(tokenizer.all_special_tokens)
29
+ tokenizer_len = len(tokenizer)
30
31
class CachedTokenizer(tokenizer.__class__):
32
@@ -41,6 +42,9 @@ def all_special_tokens(self):
41
42
def all_special_tokens_extended(self):
43
return tokenizer_all_special_tokens_extended
44
45
+ def __len__(self):
46
+ return tokenizer_len
47
+
48
CachedTokenizer.__name__ = f"Cached{tokenizer.__class__.__name__}"
49
50
tokenizer.__class__ = CachedTokenizer
0 commit comments