def setup_logging()

in src/optimum/nvidia/logging.py [0:0]


def setup_logging(verbose: bool = False, for_trtllm_logger: bool = False):
    basicConfig(format=DEFAULT_LOGGING_FMT, level=DEBUG if verbose else INFO)

    if for_trtllm_logger:
        from tensorrt_llm.logger import logger

        logger.set_level("verbose" if verbose else "info")