diff --git a/python/sglang/srt/managers/scheduler.py b/python/sglang/srt/managers/scheduler.py index 57627e55c..2b0dbae5f 100644 --- a/python/sglang/srt/managers/scheduler.py +++ b/python/sglang/srt/managers/scheduler.py @@ -2472,8 +2472,10 @@ class Scheduler( if self.profiler_decode_ct > self.profiler_target_decode_ct: if self.profile_in_progress: self.stop_profile(stage=ForwardMode.DECODE) + elif batch.forward_mode.is_idle(): + pass else: - raise RuntimeError("unsupported profile stage") + raise RuntimeError(f"unsupported profile stage: {batch.forward_mode}") else: # Check profiler if (