[diffusion] fix: Fix profiler trace missing Python stack in diffusion pipeline (#14499)

2025-12-05 12:12:35 +00:00
parent 2ce121a1c3
commit 5347732219
1 changed files with 1 additions and 3 deletions
@@ -599,9 +599,7 @@ class DenoisingStage(PipelineStage):
                active=batch.num_profiled_timesteps,
                repeat=5,
            ),
-            on_trace_ready=lambda _: torch.profiler.tensorboard_trace_handler(
+            on_trace_ready=None,
                f"./logs"
            ),
            record_shapes=True,
            with_stack=True,
        )