[Misc] Suppress duplicated logging regarding multimodal input pipeline (

vllm-project#10530) Signed-off-by: Roger Wang <[email protected]> Signed-off-by: Maxime Fournioux <[email protected]>
mfournioux · Nov 28, 2024 · fee1e44 · fee1e44
1 parent a682208
commit fee1e44
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/vllm/inputs/preprocess.py b/vllm/inputs/preprocess.py
@@ -10,7 +10,7 @@
 from vllm.multimodal.processing import MultiModalDataDict, MultiModalInputsV2
 from vllm.prompt_adapter.request import PromptAdapterRequest
 from vllm.transformers_utils.tokenizer_group import BaseTokenizerGroup
-from vllm.utils import print_warning_once
+from vllm.utils import print_info_once, print_warning_once
 
 from .data import (DecoderOnlyInputs, EncoderDecoderInputs, ProcessorInputs,
                    PromptType, SingletonInputs, SingletonPrompt, token_inputs)
@@ -212,7 +212,7 @@ def _can_process_multimodal(self) -> bool:
         # updated to use the new multi-modal processor
         can_process_multimodal = self.mm_registry.has_processor(model_config)
         if not can_process_multimodal:
-            logger.info(
+            print_info_once(
                 "Your model uses the legacy input pipeline instead of the new "
                 "multi-modal processor. Please note that the legacy pipeline "
                 "will be removed in a future release. For more details, see: "

diff --git a/vllm/utils.py b/vllm/utils.py
@@ -705,6 +705,12 @@ def create_kv_caches_with_random(
     return key_caches, value_caches
 
 
+@lru_cache
+def print_info_once(msg: str) -> None:
+    # Set the stacklevel to 2 to print the caller's line info
+    logger.info(msg, stacklevel=2)
+
+
 @lru_cache
 def print_warning_once(msg: str) -> None:
     # Set the stacklevel to 2 to print the caller's line info