[Bugfix] Fix print_warning_once's line info (vllm-project#8867)

Signed-off-by: Alvant <alvasian@yandex.ru>
compressa-ai · Oct 26, 2024 · 9e33419 · 9e33419
1 parent bfaf2b8
commit 9e33419
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/vllm/utils.py b/vllm/utils.py
@@ -744,7 +744,8 @@ def create_kv_caches_with_random(
 
 @lru_cache
 def print_warning_once(msg: str) -> None:
-    logger.warning(msg)
+    # Set the stacklevel to 2 to print the caller's line info
+    logger.warning(msg, stacklevel=2)
 
 
 @lru_cache(maxsize=None)