From 53477322192c4ba2f28b4bcac3c7752fcbb77d17 Mon Sep 17 00:00:00 2001 From: Xiaoyu Zhang <35585791+BBuf@users.noreply.github.com> Date: Fri, 5 Dec 2025 20:12:35 +0800 Subject: [PATCH] [diffusion] fix: Fix profiler trace missing Python stack in diffusion pipeline (#14499) --- .../multimodal_gen/runtime/pipelines_core/stages/denoising.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/python/sglang/multimodal_gen/runtime/pipelines_core/stages/denoising.py b/python/sglang/multimodal_gen/runtime/pipelines_core/stages/denoising.py index aae4af503..ed2f7aab8 100644 --- a/python/sglang/multimodal_gen/runtime/pipelines_core/stages/denoising.py +++ b/python/sglang/multimodal_gen/runtime/pipelines_core/stages/denoising.py @@ -599,9 +599,7 @@ class DenoisingStage(PipelineStage): active=batch.num_profiled_timesteps, repeat=5, ), - on_trace_ready=lambda _: torch.profiler.tensorboard_trace_handler( - f"./logs" - ), + on_trace_ready=None, record_shapes=True, with_stack=True, )