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 aae4af5033a8..ed2f7aab8051 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 @@ def start_profile(self, batch: Req): 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, )