Fix type error

vllm-project · Aug 28, 2024 · 5a97c5b · 5a97c5b
1 parent 61ea59c
commit 5a97c5b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/vllm/transformers_utils/tokenizers/mistral.py b/vllm/transformers_utils/tokenizers/mistral.py
@@ -52,12 +52,13 @@ def __init__(self, tokenizer: PublicMistralTokenizer) -> None:
         assert isinstance(self.tokenizer,
                           (Tekkenizer, SentencePieceTokenizer)), type(
                               self.tokenizer)
-        self._is_tekken = isinstance(self.tokenizer, Tekkenizer)
 
-        if self._is_tekken:
+        if (is_tekken := isinstance(self.tokenizer, Tekkenizer)):
             # Make sure special tokens will not raise
             self.tokenizer.special_token_policy = SpecialTokenPolicy.IGNORE
 
+        self._is_tekken = is_tekken
+
         # the following attributes are set to fit VLLM's design
         self.is_fast = True
         self.chat_template = True