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

2025-05-25 22:36:02 -07:00
parent 0ca1811715
commit 8c7279c24e
1 changed files with 1 additions and 0 deletions
@@ -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")