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

e2f6f26e · Tyler Michael Smith · GitHub · b28d2104 · e2f6f26e
Unverified Commit e2f6f26e authored Sep 26, 2024 by Tyler Michael Smith Committed by GitHub Sep 26, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

vllm/utils.py vllm/utils.py +2 -1

No files found.
--- 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)