File tree Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Original file line number Diff line number Diff line change 23
23
zmq_socket_ctx )
24
24
from vllm .v1 .core .kv_cache_utils import (get_kv_cache_config ,
25
25
unify_kv_cache_configs )
26
+ from vllm .v1 .core .sched .interface import SchedulerInterface
26
27
from vllm .v1 .core .sched .output import SchedulerOutput
27
28
from vllm .v1 .core .sched .scheduler import Scheduler as V1Scheduler
28
29
from vllm .v1 .engine import (EngineCoreOutputs , EngineCoreRequest ,
@@ -84,7 +85,7 @@ def __init__(
84
85
"compatibility may not be maintained." ,
85
86
vllm_config .scheduler_config .scheduler_cls )
86
87
87
- self .scheduler = Scheduler (
88
+ self .scheduler : SchedulerInterface = Scheduler (
88
89
scheduler_config = vllm_config .scheduler_config ,
89
90
model_config = vllm_config .model_config ,
90
91
cache_config = vllm_config .cache_config ,
You can’t perform that action at this time.
0 commit comments