File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 26
26
from vllm .engine .output_processor .stop_checker import StopChecker
27
27
from vllm .engine .output_processor .util import create_output_by_sequence_group
28
28
from vllm .executor .executor_base import ExecutorBase
29
- from vllm .executor .gpu_executor import GPUExecutorAsync
29
+ from vllm .executor .gpu_executor import GPUExecutor
30
30
from vllm .executor .ray_utils import initialize_ray_cluster
31
31
from vllm .inputs import (INPUT_REGISTRY , EncoderDecoderLLMInputs ,
32
32
InputRegistry , LLMInputs , PromptInputs ,
@@ -1965,13 +1965,13 @@ def check_health(self) -> None:
1965
1965
self .model_executor .check_health ()
1966
1966
1967
1967
def start_profile (self ) -> None :
1968
- if isinstance (self .model_executor , GPUExecutorAsync ) :
1968
+ if type (self .model_executor ) == GPUExecutor :
1969
1969
self .model_executor .start_profile ()
1970
1970
else :
1971
1971
self .model_executor ._run_workers ("start_profile" )
1972
1972
1973
1973
def stop_profile (self ) -> None :
1974
- if isinstance (self .model_executor , GPUExecutorAsync ) :
1974
+ if type (self .model_executor ) == GPUExecutor :
1975
1975
self .model_executor .stop_profile ()
1976
1976
else :
1977
1977
self .model_executor ._run_workers ("stop_profile" )
You can’t perform that action at this time.
0 commit comments