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 2323 zmq_socket_ctx )
2424from vllm .v1 .core .kv_cache_utils import (get_kv_cache_config ,
2525 unify_kv_cache_configs )
26+ from vllm .v1 .core .sched .interface import SchedulerInterface
2627from vllm .v1 .core .sched .output import SchedulerOutput
2728from vllm .v1 .core .sched .scheduler import Scheduler as V1Scheduler
2829from vllm .v1 .engine import (EngineCoreOutputs , EngineCoreRequest ,
@@ -84,7 +85,7 @@ def __init__(
8485 "compatibility may not be maintained." ,
8586 vllm_config .scheduler_config .scheduler_cls )
8687
87- self .scheduler = Scheduler (
88+ self .scheduler : SchedulerInterface = Scheduler (
8889 scheduler_config = vllm_config .scheduler_config ,
8990 model_config = vllm_config .model_config ,
9091 cache_config = vllm_config .cache_config ,
You can’t perform that action at this time.
0 commit comments