diff --git a/tpu_inference/platforms/tpu_platform.py b/tpu_inference/platforms/tpu_platform.py index c85070da44..1c9478c9f4 100644 --- a/tpu_inference/platforms/tpu_platform.py +++ b/tpu_inference/platforms/tpu_platform.py @@ -247,9 +247,8 @@ def supports_v1(cls, model_config: ModelConfig) -> bool: @classmethod def validate_request( cls, - prompt: PromptType, - params: Union["SamplingParams", PoolingParams], processed_inputs: ProcessorInputs, + params: Union["SamplingParams", PoolingParams], ) -> None: """Raises if this request is unsupported on this platform""" from vllm.sampling_params import SamplingParams, SamplingType