diff --git a/python/sglang/srt/model_executor/cuda_graph_runner.py b/python/sglang/srt/model_executor/cuda_graph_runner.py index 55a23e333..b0b2ede6d 100644 --- 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)