Pipe customized_info through CudaGraphRunner output (#17088)

2026-01-19 15:48:49 -08:00
parent 7e40d52635
commit 09a9d214f7
1 changed files with 1 additions and 0 deletions
--- a/python/sglang/srt/model_executor/cuda_graph_runner.py
+++ b/python/sglang/srt/model_executor/cuda_graph_runner.py
@@ -878,6 +878,7 @@ class CudaGraphRunner:
                    if output.hidden_states is not None
                    else None
                ),
+                customized_info=output.customized_info,
            )
        else:
            assert isinstance(output, PPProxyTensors)