Skip to content

Commit

Permalink
reformat based on 'bash format.sh'
Browse files Browse the repository at this point in the history
Signed-off-by: Sophie du Couédic <sop@zurich.ibm.com>
  • Loading branch information
sducouedic committed Jan 31, 2025
1 parent ca4a8e7 commit 4e606a6
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion vllm/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -1487,7 +1487,8 @@ def __post_init__(self) -> None:

self.chunked_prefill_enabled = self.enable_chunked_prefill
from vllm.platforms import current_platform
self.spyre_scheduling_enabled = current_platform.get_device_name() == "spyre"
self.spyre_scheduling_enabled = current_platform.get_device_name(
) == "spyre"
if self.spyre_scheduling_enabled:
# load warmup shapes and sort by "speed"
wup_prompt_lens = envs.VLLM_SPYRE_WARMUP_PROMPT_LENS or []
Expand Down

0 comments on commit 4e606a6

Please sign in to comment.