[feat] Added num_gpu_blocks limit gpu blocks.

2025-12-10 20:17:42 +08:00
parent 01f19ee4a6
commit 0a247ccb1b
7 changed files with 150 additions and 9 deletions
--- a/nanovllm/utils/logger.py
+++ b/nanovllm/utils/logger.py
@@ -83,9 +83,9 @@ def _setup_logger() -> logging.Logger:
    # Check if terminal supports colors
    use_colors = hasattr(sys.stderr, "isatty") and sys.stderr.isatty()

-    # Format: [TIME] [LEVEL] [MODULE] message
+    # Format: [TIME] [LEVEL] [FILE:LINE] message
    formatter = ColoredFormatter(
-        fmt="[%(asctime)s] [%(levelname)s] [%(name)s] %(message)s",
+        fmt="[%(asctime)s] [%(levelname)s] [%(filename)s:%(lineno)d] %(message)s",
        datefmt="%H:%M:%S",
        use_colors=use_colors,
    )