only profile driver worker stop/start

sfc-gh-mkeralapura · Sep 5, 2024 · b6f8764 · b6f8764
1 parent 6f3061b
commit b6f8764
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vllm/worker/worker.py b/vllm/worker/worker.py
@@ -135,12 +135,12 @@ def __init__(
             self.profiler = None
 
     def start_profile(self):
-        if self.profiler is None:
+        if self.is_driver_worker and self.profiler is None:
             raise RuntimeError("Profiler is not enabled.")
         self.profiler.start()
 
     def stop_profile(self):
-        if self.profiler is None:
+        if self.is_driver_worker and self.profiler is None:
             raise RuntimeError("Profiler is not enabled.")
         self.profiler.stop()