diff --git a/vllm_omni/entrypoints/omni_stage.py b/vllm_omni/entrypoints/omni_stage.py index 0c4d61911..5fc05dfba 100644 --- a/vllm_omni/entrypoints/omni_stage.py +++ b/vllm_omni/entrypoints/omni_stage.py @@ -893,7 +893,7 @@ async def _stage_worker_async( _rx_bytes_by_rid[rid] = int(_rx_metrics.get("rx_transfer_bytes", 0)) sampling_params = task["sampling_params"] - _logging.getLogger(__name__).debug("[Stage-%s] Received batch size=1, request_ids=%d", stage_id, rid) + _logging.getLogger(__name__).debug("[Stage-%s] Received batch size=1, request_ids=%s", stage_id, rid) print("--------------------------------", flush=True) print(f"[Stage-{stage_id}] Received batch size=1, request_ids={rid}", flush=True) print("--------------------------------", flush=True) diff --git a/vllm_omni/logger.py b/vllm_omni/logger.py new file mode 100644 index 000000000..1bcbc027b --- /dev/null +++ b/vllm_omni/logger.py @@ -0,0 +1,22 @@ +import logging + +from vllm.logger import init_logger + + +def _configure_vllm_omni_root_logger(): + """ + Configure the root logger for vllm_omni to propagate to vllm's root logger. + """ + vllm_root = logging.getLogger("vllm") + vllm_omni_root = logging.getLogger("vllm_omni") + vllm_omni_root.handlers = [] + + vllm_omni_root.parent = vllm_root + + vllm_omni_root.propagate = True + + vllm_omni_root.setLevel(logging.NOTSET) + + +_configure_vllm_omni_root_logger() +init_logger(__name__) diff --git a/vllm_omni/patch.py b/vllm_omni/patch.py index 6cd344de8..687ff5186 100644 --- a/vllm_omni/patch.py +++ b/vllm_omni/patch.py @@ -9,6 +9,7 @@ from vllm.v1.engine import EngineCoreOutputs as _OriginalEngineCoreOutputs from vllm.v1.engine import EngineCoreRequest as _OriginalEngineCoreRequest from vllm.v1.request import Request as _OriginalRequest +import vllm_omni.logger # noqa: F401 from vllm_omni.engine import OmniEngineCoreOutput, OmniEngineCoreOutputs, OmniEngineCoreRequest from vllm_omni.inputs.data import OmniTokensPrompt from vllm_omni.model_executor.layers.mrope import MRotaryEmbedding