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

2025-05-26 13:36:02 +08:00
parent 0ca1811715
commit 8c7279c24e
1 changed files with 1 additions and 0 deletions
--- 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")