Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
203d4f82
Unverified
Commit
203d4f82
authored
Mar 29, 2024
by
youkaichao
Committed by
GitHub
Mar 29, 2024
Browse files
[Core][Bugfix] cache len of tokenizer (#3741)
parent
991143cf
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
0 deletions
+4
-0
vllm/transformers_utils/tokenizer.py
vllm/transformers_utils/tokenizer.py
+4
-0
No files found.
vllm/transformers_utils/tokenizer.py
View file @
203d4f82
...
@@ -26,6 +26,7 @@ def get_cached_tokenizer(
...
@@ -26,6 +26,7 @@ def get_cached_tokenizer(
tokenizer_all_special_tokens_extended
=
(
tokenizer_all_special_tokens_extended
=
(
tokenizer
.
all_special_tokens_extended
)
tokenizer
.
all_special_tokens_extended
)
tokenizer_all_special_tokens
=
set
(
tokenizer
.
all_special_tokens
)
tokenizer_all_special_tokens
=
set
(
tokenizer
.
all_special_tokens
)
tokenizer_len
=
len
(
tokenizer
)
class
CachedTokenizer
(
tokenizer
.
__class__
):
class
CachedTokenizer
(
tokenizer
.
__class__
):
...
@@ -41,6 +42,9 @@ def get_cached_tokenizer(
...
@@ -41,6 +42,9 @@ def get_cached_tokenizer(
def
all_special_tokens_extended
(
self
):
def
all_special_tokens_extended
(
self
):
return
tokenizer_all_special_tokens_extended
return
tokenizer_all_special_tokens_extended
def
__len__
(
self
):
return
tokenizer_len
CachedTokenizer
.
__name__
=
f
"Cached
{
tokenizer
.
__class__
.
__name__
}
"
CachedTokenizer
.
__name__
=
f
"Cached
{
tokenizer
.
__class__
.
__name__
}
"
tokenizer
.
__class__
=
CachedTokenizer
tokenizer
.
__class__
=
CachedTokenizer
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment