Sync changes on io_struct.py and deterministic ops (#11498)

2025-10-12 16:03:10 -07:00
parent 0aa65f94f1
commit 2ac46e94ef
11 changed files with 73 additions and 25 deletions
@@ -920,7 +920,8 @@ class SchedulerOutputProcessorMixin:
                    input_token_ids_logprobs_idx,
                    output_token_ids_logprobs_val,
                    output_token_ids_logprobs_idx,
-                    output_hidden_states,
+                    output_token_entropy_val=None,
+                    output_hidden_states=output_hidden_states,
                    rids=rids,
                    placeholder_tokens_idx=None,
                    placeholder_tokens_val=None,