Update vllm/worker/multi_step_model_runner.py

Co-authored-by: afeldman-nm <[email protected]>

vllm/worker/multi_step_model_runner.py

@@ -412,7 +412,7 @@ def vocab_size(self) -> int:
 def _pythonize_sampler_output(
         model_input: StatefulModelInput, output: SamplerOutput,
         pinned_sampled_token_buffer: torch.Tensor,
-        sampled_token_ids: torch.Tensor) -> SamplerOutput:
+        sampled_token_ids: torch.Tensor) -> None:
     """ This function is only called when the output tensors are ready. 
     See ModelOutput
     """