Skip to content

Commit f98a492

Browse files
authored
[V1][Core] Remove unused speculative config from scheduler (#15818)
Signed-off-by: Mark McLoughlin <[email protected]>
1 parent d4bfc23 commit f98a492

File tree

3 files changed

+1
-6
lines changed

3 files changed

+1
-6
lines changed

tests/v1/core/test_scheduler.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,6 @@ def create_scheduler(
7171
scheduler_config,
7272
model_config,
7373
cache_config,
74-
speculative_config=None,
7574
lora_config=None,
7675
log_stats=True,
7776
structured_output_manager=StructuredOutputManager(vllm_config),

vllm/v1/core/sched/scheduler.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,7 @@
77
from collections.abc import Iterable
88
from typing import Optional, Union
99

10-
from vllm.config import (CacheConfig, LoRAConfig, ModelConfig, SchedulerConfig,
11-
SpeculativeConfig)
10+
from vllm.config import CacheConfig, LoRAConfig, ModelConfig, SchedulerConfig
1211
from vllm.logger import init_logger
1312
from vllm.multimodal import MULTIMODAL_REGISTRY, MultiModalRegistry
1413
from vllm.v1.core.encoder_cache_manager import (EncoderCacheManager,
@@ -36,7 +35,6 @@ def __init__(
3635
model_config: ModelConfig,
3736
cache_config: CacheConfig,
3837
lora_config: Optional[LoRAConfig],
39-
speculative_config: Optional[SpeculativeConfig],
4038
structured_output_manager: StructuredOutputManager,
4139
mm_registry: MultiModalRegistry = MULTIMODAL_REGISTRY,
4240
include_finished_set: bool = False,
@@ -45,7 +43,6 @@ def __init__(
4543
self.scheduler_config = scheduler_config
4644
self.cache_config = cache_config
4745
self.lora_config = lora_config
48-
self.speculative_config = speculative_config
4946
self.log_stats = log_stats
5047
self.structured_output_manager = structured_output_manager
5148

vllm/v1/engine/core.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,6 @@ def __init__(
9595
model_config=vllm_config.model_config,
9696
cache_config=vllm_config.cache_config,
9797
lora_config=vllm_config.lora_config,
98-
speculative_config=vllm_config.speculative_config,
9998
include_finished_set=vllm_config.parallel_config.data_parallel_size
10099
> 1,
101100
log_stats=self.log_stats,

0 commit comments

Comments
 (0)