Update vllm/engine/llm_engine.py

vllm-project · Jul 24, 2024 · d097cba · d097cba
1 parent 4c9ef39
commit d097cba
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,