Set local logging level via env variable (vllm-project#2774)

xjpang · Feb 22, 2024 · a62ec2d · a62ec2d
1 parent 4a8ccf4
commit a62ec2d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/vllm/logger.py b/vllm/logger.py
@@ -3,6 +3,7 @@
 """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"
@@ -50,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(logging.DEBUG)
+    logger.setLevel(os.getenv("LOG_LEVEL", "DEBUG"))
     logger.addHandler(_default_handler)
     logger.propagate = False
     return logger