From c71cc535cae3114a1e8195350c52eff76f1c97cd Mon Sep 17 00:00:00 2001 From: LukasGardberg Date: Mon, 5 Feb 2024 22:07:49 +0100 Subject: [PATCH 1/2] set log level via env --- vllm/logger.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/vllm/logger.py b/vllm/logger.py index 24d4f0ec1ae0a..ddae3e58beb61 100644 --- a/vllm/logger.py +++ b/vllm/logger.py @@ -3,9 +3,11 @@ """Logging configuration for vLLM.""" import logging import sys +import os _FORMAT = "%(levelname)s %(asctime)s %(filename)s:%(lineno)d] %(message)s" _DATE_FORMAT = "%m-%d %H:%M:%S" +_LOG_LEVEL = os.getenv("LOG_LEVEL", "INFO") class NewLineFormatter(logging.Formatter): @@ -32,7 +34,7 @@ def _setup_logger(): if _default_handler is None: _default_handler = logging.StreamHandler(sys.stdout) _default_handler.flush = sys.stdout.flush # type: ignore - _default_handler.setLevel(logging.INFO) + _default_handler.setLevel(_LOG_LEVEL) _root_logger.addHandler(_default_handler) fmt = NewLineFormatter(_FORMAT, datefmt=_DATE_FORMAT) _default_handler.setFormatter(fmt) @@ -50,7 +52,7 @@ def _setup_logger(): def init_logger(name: str): # Use the same settings as above for root logger logger = logging.getLogger(name) - logger.setLevel(logging.DEBUG) + logger.setLevel(_LOG_LEVEL) logger.addHandler(_default_handler) logger.propagate = False return logger From dac5f258a355978b576971c8cc33dd1193b5e017 Mon Sep 17 00:00:00 2001 From: LukasGardberg Date: Mon, 5 Feb 2024 22:19:52 +0100 Subject: [PATCH 2/2] only set local --- vllm/logger.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/vllm/logger.py b/vllm/logger.py index ddae3e58beb61..530494ae66925 100644 --- a/vllm/logger.py +++ b/vllm/logger.py @@ -7,7 +7,6 @@ _FORMAT = "%(levelname)s %(asctime)s %(filename)s:%(lineno)d] %(message)s" _DATE_FORMAT = "%m-%d %H:%M:%S" -_LOG_LEVEL = os.getenv("LOG_LEVEL", "INFO") class NewLineFormatter(logging.Formatter): @@ -34,7 +33,7 @@ def _setup_logger(): if _default_handler is None: _default_handler = logging.StreamHandler(sys.stdout) _default_handler.flush = sys.stdout.flush # type: ignore - _default_handler.setLevel(_LOG_LEVEL) + _default_handler.setLevel(logging.INFO) _root_logger.addHandler(_default_handler) fmt = NewLineFormatter(_FORMAT, datefmt=_DATE_FORMAT) _default_handler.setFormatter(fmt) @@ -52,7 +51,7 @@ def _setup_logger(): def init_logger(name: str): # Use the same settings as above for root logger logger = logging.getLogger(name) - logger.setLevel(_LOG_LEVEL) + logger.setLevel(os.getenv("LOG_LEVEL", "DEBUG")) logger.addHandler(_default_handler) logger.propagate = False return logger