Skip to content

Commit 29d9e6b

Browse files
committed
Save the vllm tokenizer adapted state
1 parent e16d986 commit 29d9e6b

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

outlines/serve/vllm.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,9 @@ def _adapt_tokenizer(tokenizer):
4848
Llama's tokenizer to be able to compile FSMs for this model.
4949
5050
"""
51+
if getattr(tokenizer, "_outlines_adapted", False):
52+
return tokenizer
53+
5154
tokenizer.vocabulary = tokenizer.get_vocab()
5255
tokenizer.special_tokens = set(tokenizer.all_special_tokens)
5356

@@ -72,6 +75,7 @@ def new_decoder(inp_tokens: List[int]) -> List[str]:
7275

7376
tokenizer.convert_token_to_string = convert_token_to_string
7477
tokenizer.decode = change_decoder(tokenizer.decode)
78+
setattr(tokenizer, "_outlines_adapted", True)
7579

7680
return tokenizer
7781

0 commit comments

Comments
 (0)