[MRV2] Validate MRV2 entrypoint logits processors (#57728)
Signed-off-by: Taneem Ibrahim <taneem.ibrahim@gmail.com>
Taneem Ibrahim committed
Sep 24, 2026 at 05:07 UTC
bbd7c24d6e365dbad98543701a6bde363f1d8aa3
2 files changed
+6
-3
vllm/v1/engine/input_processor.py
+3
@@ -97,6 +97,9 @@ class InputProcessor:
97
params validator for the active model runner."""
98
custom_logitsprocs = self.model_config.logits_processors
99
if self.vllm_config.use_v2_model_runner:
100
+ if self.model_config.runner_type == "pooling" and not custom_logitsprocs:
101
+ return lambda _: None
102
+
103
from vllm.v1.worker.gpu.sample.logits_processor import (
104
build_custom_logits_processors_params_validator,
105
)
vllm/v1/worker/gpu/sample/logits_processor/loader.py
+3
-3
@@ -196,10 +196,10 @@ def build_custom_logits_processors_params_validator(
196
RuntimeError: if an FQCN fails to import.
197
198
"""
199
- if not custom_logitsprocs:
200
- return lambda _: None
199
+ classes = _cached_load_v2_logitsprocs(tuple(custom_logitsprocs or ()))
200
202
- classes = _cached_load_v2_logitsprocs(tuple(custom_logitsprocs))
201
+ if not classes:
202
+ return lambda _: None
203
204
def validate_params(sampling_params: "SamplingParams") -> None:
205
for cls in classes: