Skip to content

Commit b870182

Browse files
committed
chore: fix ci lint
Signed-off-by: elijah <[email protected]>
1 parent 971102d commit b870182

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

benchmarks/backend_request_func.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -423,7 +423,7 @@ def get_tokenizer(
423423
**kwargs,
424424
) -> Union[PreTrainedTokenizer, PreTrainedTokenizerFast]:
425425
if pretrained_model_name_or_path is not None and not os.path.exists(
426-
pretrained_model_name_or_path):
426+
pretrained_model_name_or_path):
427427
pretrained_model_name_or_path = get_model(
428428
pretrained_model_name_or_path)
429429
if tokenizer_mode == "slow":
@@ -433,12 +433,13 @@ def get_tokenizer(
433433
kwargs["use_fast"] = False
434434
if tokenizer_mode == "mistral":
435435
try:
436-
from vllm.transformers_utils.tokenizer import (
437-
MistralTokenizer)
438-
except ImportError:
436+
from vllm.transformers_utils.tokenizer import MistralTokenizer
437+
except ImportError as e:
439438
raise ImportError(
440439
"MistralTokenizer requires vllm package.\n"
441-
"Please install it with `pip install vllm` to use mistral tokenizer mode.")
440+
"Please install it with `pip install vllm` "
441+
"to use mistral tokenizer mode."
442+
) from e
442443
return MistralTokenizer.from_pretrained(
443444
str(pretrained_model_name_or_path))
444445
else:

0 commit comments

Comments
 (0)