Skip to content

GitLab

Unverified Commit 8c7279c2 authored by fzyzcjy's avatar fzyzcjy Committed by GitHub
Browse files

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

parent 0ca18117
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
python/sglang/srt/managers/scheduler.py
View file @ 8c7279c2
...@@ -2203,6 +2203,7 @@ class Scheduler( ...@@ -2203,6 +2203,7 @@ class Scheduler(
self.torch_profiler = None self.torch_profiler = None
self.torch_profiler_output_dir = None self.torch_profiler_output_dir = None
self.profiler_activities = None self.profiler_activities = None
self.profiler_target_forward_ct = None
return ProfileReqOutput(success=True, message="Succeeded") return ProfileReqOutput(success=True, message="Succeeded")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment