diff --git a/python/sglang/srt/managers/scheduler_components/profiler_manager.py b/python/sglang/srt/managers/scheduler_components/profiler_manager.py index a4c10fc7e..1ebeeec4f 100644 --- a/python/sglang/srt/managers/scheduler_components/profiler_manager.py +++ b/python/sglang/srt/managers/scheduler_components/profiler_manager.py @@ -144,7 +144,7 @@ class SchedulerProfilerManager: self.profiler_start_forward_ct + num_steps ) else: - self.profiler_target_forward_ct = self.get_forward_ct() + num_steps + self.profiler_target_forward_ct = self.get_forward_ct() + num_steps + 1 # The caller will be notified when reaching profiler_target_forward_ct else: self.profiler_target_forward_ct = None