Skip to content

GitLab

Unverified Commit e2f6f26e authored by Tyler Michael Smith's avatar Tyler Michael Smith Committed by GitHub
Browse files

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

parent b28d2104
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
vllm/utils.py
View file @ e2f6f26e
...@@ -744,7 +744,8 @@ def create_kv_caches_with_random( ...@@ -744,7 +744,8 @@ def create_kv_caches_with_random(
@lru_cache @lru_cache
def print_warning_once(msg: str) -> None: 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) @lru_cache(maxsize=None)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment