Fix profiling will crash the server when using num_steps (#6586)

8c7279c2 · fzyzcjy · GitHub · 0ca18117 · 8c7279c2
Unverified Commit 8c7279c2 authored May 26, 2025 by fzyzcjy Committed by GitHub May 25, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

python/sglang/srt/managers/scheduler.py python/sglang/srt/managers/scheduler.py +1 -0

No files found.
--- a/python/sglang/srt/managers/scheduler.py
+++ b/python/sglang/srt/managers/scheduler.py
@@ -2203,6 +2203,7 @@ class Scheduler(
        self.torch_profiler = None
        self.torch_profiler_output_dir = None
        self.profiler_activities = None
+        self.profiler_target_forward_ct = None

        return ProfileReqOutput(success=True, message="Succeeded")