xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-11 02:45:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Bugfix] Print warnings related to mistral_common tokenizer only once (#9468)

Browse Source
This commit is contained in:
sasha0552 2024-10-17 21:44:20 +00:00 committed by GitHub
parent d65049daab
commit d615b5c9f8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
vllm/entrypoints/chat_utils.py
View File

@ -33,6 +33,7 @@ from vllm.multimodal.utils import (async_get_and_parse_audio,
async_get_and_parse_image, async_get_and_parse_image,
get_and_parse_audio, get_and_parse_image) get_and_parse_audio, get_and_parse_image)
from vllm.transformers_utils.tokenizer import AnyTokenizer, MistralTokenizer from vllm.transformers_utils.tokenizer import AnyTokenizer, MistralTokenizer
from vllm.utils import print_warning_once
logger = init_logger(__name__) logger = init_logger(__name__)
@ -564,14 +565,14 @@ def apply_mistral_chat_template(
**kwargs: Any, **kwargs: Any,
) -> List[int]: ) -> List[int]:
if chat_template is not None: if chat_template is not None:
logger.warning( print_warning_once(
"'chat_template' cannot be overridden for mistral tokenizer.") "'chat_template' cannot be overridden for mistral tokenizer.")
if "add_generation_prompt" in kwargs: if "add_generation_prompt" in kwargs:
logger.warning( print_warning_once(
"'add_generation_prompt' is not supported for mistral tokenizer, " "'add_generation_prompt' is not supported for mistral tokenizer, "
"so it will be ignored.") "so it will be ignored.")
if "continue_final_message" in kwargs: if "continue_final_message" in kwargs:
logger.warning( print_warning_once(
"'continue_final_message' is not supported for mistral tokenizer, " "'continue_final_message' is not supported for mistral tokenizer, "
"so it will be ignored.") "so it will be ignored.")