forked from EngineX-Cambricon/enginex-mlu370-vllm
18 lines
484 B
Python
18 lines
484 B
Python
import logging
|
|
from logging import Logger
|
|
|
|
def init_logger(name: str) -> Logger:
|
|
"""Initialize loggers for benchmarks module,
|
|
and keep the configuration consistent with the vllm module"""
|
|
|
|
logger = logging.getLogger(name)
|
|
|
|
vllm_logger = logging.Logger.manager.loggerDict.get('vllm', None)
|
|
if vllm_logger:
|
|
logger.setLevel(vllm_logger.level)
|
|
logger.propagate = vllm_logger.propagate
|
|
logger.handlers = vllm_logger.handlers
|
|
|
|
return logger
|
|
|