format

vllm-project · Aug 19, 2024 · 5fac4a1 · 5fac4a1
1 parent a2b3c7e
commit 5fac4a1
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/vllm/worker/multi_step_model_runner.py b/vllm/worker/multi_step_model_runner.py
@@ -409,10 +409,10 @@ def vocab_size(self) -> int:
         return self._base_model_runner.vocab_size
 
 
-def _pythonize_sampler_output(
-        model_input: StatefulModelInput, output: SamplerOutput,
-        pinned_sampled_token_buffer: torch.Tensor,
-        sampled_token_ids: torch.Tensor) -> None:
+def _pythonize_sampler_output(model_input: StatefulModelInput,
+                              output: SamplerOutput,
+                              pinned_sampled_token_buffer: torch.Tensor,
+                              sampled_token_ids: torch.Tensor) -> None:
     """ This function is only called when the output tensors are ready. 
     See ModelOutput
     """