[Bugfix] Fix non-default argument error

Fix bug due to vllm-project/vllm#24278

Signed-off-by: Kyuyeun Kim <kyuyeunk@google.com>

Author: kyuyeunk

tpu_commons/runner/input_batch_jax.py

@@ -23,7 +23,7 @@
 @dataclass
 class CachedRequestState(NewRequestData):
 
-    output_token_ids: list[int]
+    output_token_ids: list[int] = None
     generator: Optional[Any] = None
     mrope_positions: Optional[jax.Array] = None
     mrope_position_delta: Optional[int] = None