Pipe customized_info through CudaGraphRunner output (#17088)
This commit is contained in:
@@ -878,6 +878,7 @@ class CudaGraphRunner:
|
|||||||
if output.hidden_states is not None
|
if output.hidden_states is not None
|
||||||
else None
|
else None
|
||||||
),
|
),
|
||||||
|
customized_info=output.customized_info,
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
assert isinstance(output, PPProxyTensors)
|
assert isinstance(output, PPProxyTensors)
|
||||||
|
|||||||
Reference in New Issue
Block a user