fix start/stop

sfc-gh-mkeralapura · Sep 6, 2024 · 47e0252 · 47e0252
1 parent 098f90f
commit 47e0252
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/vllm/worker/worker.py b/vllm/worker/worker.py
@@ -135,15 +135,15 @@ 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.")
-        if self.is_driver_worker:
+        elif self.profiler is not None:
             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.")
-        if self.is_driver_worker:
+        elif self.profiler is not None:
             self.profiler.stop()
 
     def _is_encoder_decoder_model(self):