Update vllm/engine/llm_engine.py

vllm-project · Jul 23, 2024 · 184f772 · 184f772
1 parent d3940ea
commit 184f772
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/vllm/engine/llm_engine.py b/vllm/engine/llm_engine.py
@@ -948,9 +948,10 @@ def do_log_stats(
             scheduler_outputs: Optional[SchedulerOutputs] = None,
             model_output: Optional[List[SamplerOutput]] = None) -> None:
         """Forced log when no requests active."""
-        stats = self._get_stats(scheduler_outputs, model_output)
-        for logger in self.stat_loggers.values():
-            logger.log(stats)
+        if self.log_stats:
+            stats = self._get_stats(scheduler_outputs, model_output)
+            for logger in self.stat_loggers.values():
+                logger.log(stats)
 
     def _get_stats(
             self,