Skip to content

Commit e132dde

Browse files
change to type
1 parent eaf0c01 commit e132dde

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

vllm/engine/llm_engine.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626
from vllm.engine.output_processor.stop_checker import StopChecker
2727
from vllm.engine.output_processor.util import create_output_by_sequence_group
2828
from vllm.executor.executor_base import ExecutorBase
29-
from vllm.executor.gpu_executor import GPUExecutorAsync
29+
from vllm.executor.gpu_executor import GPUExecutor
3030
from vllm.executor.ray_utils import initialize_ray_cluster
3131
from vllm.inputs import (INPUT_REGISTRY, EncoderDecoderLLMInputs,
3232
InputRegistry, LLMInputs, PromptInputs,
@@ -1965,13 +1965,13 @@ def check_health(self) -> None:
19651965
self.model_executor.check_health()
19661966

19671967
def start_profile(self) -> None:
1968-
if isinstance(self.model_executor, GPUExecutorAsync):
1968+
if type(self.model_executor) == GPUExecutor:
19691969
self.model_executor.start_profile()
19701970
else:
19711971
self.model_executor._run_workers("start_profile")
19721972

19731973
def stop_profile(self) -> None:
1974-
if isinstance(self.model_executor, GPUExecutorAsync):
1974+
if type(self.model_executor) == GPUExecutor:
19751975
self.model_executor.stop_profile()
19761976
else:
19771977
self.model_executor._run_workers("stop_profile")

0 commit comments

Comments
 (0)