From 181b27d8813e6a92de4f38cecfa24914e652588a Mon Sep 17 00:00:00 2001 From: Antoni Baum Date: Tue, 20 Feb 2024 14:38:55 -0800 Subject: [PATCH] Make vLLM logging formatting optional (#2877) --- vllm/logger.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/vllm/logger.py b/vllm/logger.py index 530494ae6692..d25fcef9ba2e 100644 --- a/vllm/logger.py +++ b/vllm/logger.py @@ -5,6 +5,8 @@ import logging import sys import os +VLLM_CONFIGURE_LOGGING = int(os.getenv("VLLM_CONFIGURE_LOGGING", "1")) + _FORMAT = "%(levelname)s %(asctime)s %(filename)s:%(lineno)d] %(message)s" _DATE_FORMAT = "%m-%d %H:%M:%S" @@ -45,13 +47,15 @@ def _setup_logger(): # The logger is initialized when the module is imported. # This is thread-safe as the module is only imported once, # guaranteed by the Python GIL. -_setup_logger() +if VLLM_CONFIGURE_LOGGING: + _setup_logger() def init_logger(name: str): # Use the same settings as above for root logger logger = logging.getLogger(name) logger.setLevel(os.getenv("LOG_LEVEL", "DEBUG")) - logger.addHandler(_default_handler) - logger.propagate = False + if VLLM_CONFIGURE_LOGGING: + logger.addHandler(_default_handler) + logger.propagate = False return logger