Fix error with deploy test

Signed-off-by: Thomas Parnell <[email protected]>
IBM · Jan 17, 2025 · d19ff3d · d19ff3d
1 parent 43b5c48
commit d19ff3d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/vllm/core/scheduler.py b/vllm/core/scheduler.py
@@ -919,8 +919,9 @@ def _schedule_prefills(
         ignored_seq_groups: List[SequenceGroup] = []
         seq_groups: List[ScheduledSequenceGroup] = []
 
-        applicable_spyre_warmup_shapes = list(
-            self.scheduler_config.spyre_warmup_shapes)
+        if self.scheduler_config.spyre_scheduling_enabled:
+            applicable_spyre_warmup_shapes = list(
+                self.scheduler_config.spyre_warmup_shapes)
 
         waiting_queue = self.waiting