change to type

vllm-project · Sep 12, 2024 · ec866f3 · ec866f3
1 parent 0d0325e
commit ec866f3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/vllm/engine/llm_engine.py b/vllm/engine/llm_engine.py
@@ -26,7 +26,7 @@
 from vllm.engine.output_processor.stop_checker import StopChecker
 from vllm.engine.output_processor.util import create_output_by_sequence_group
 from vllm.executor.executor_base import ExecutorBase
-from vllm.executor.gpu_executor import GPUExecutorAsync
+from vllm.executor.gpu_executor import GPUExecutor
 from vllm.executor.ray_utils import initialize_ray_cluster
 from vllm.inputs import (INPUT_REGISTRY, EncoderDecoderLLMInputs,
                          InputRegistry, LLMInputs, PromptInputs,
@@ -1965,13 +1965,13 @@ def check_health(self) -> None:
         self.model_executor.check_health()
 
     def start_profile(self) -> None:
-        if isinstance(self.model_executor, GPUExecutorAsync):
+        if type(self.model_executor) == GPUExecutor:
             self.model_executor.start_profile()
         else:
             self.model_executor._run_workers("start_profile")
 
     def stop_profile(self) -> None:
-        if isinstance(self.model_executor, GPUExecutorAsync):
+        if type(self.model_executor) == GPUExecutor:
             self.model_executor.stop_profile()
         else:
             self.model_executor._run_workers("stop_profile")