From 43eea2953ba23b80cc174bd24c9ce4886e6459f3 Mon Sep 17 00:00:00 2001 From: Woosuk Kwon Date: Sat, 26 Apr 2025 11:10:14 -0700 Subject: [PATCH] [Minor] Fix lint error in main branch (#17233) Signed-off-by: Woosuk Kwon --- vllm/model_executor/layers/rotary_embedding.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vllm/model_executor/layers/rotary_embedding.py b/vllm/model_executor/layers/rotary_embedding.py index 340ff94cd94c3..c5970c71c539f 100644 --- a/vllm/model_executor/layers/rotary_embedding.py +++ b/vllm/model_executor/layers/rotary_embedding.py @@ -1329,8 +1329,8 @@ class MRotaryEmbedding(RotaryEmbedding): audio_llm_pos_ids_list = (torch.arange( min(t_ntoken_per_chunk, pure_audio_len - added_audio_len)).expand(3, -1) + - audio_start_idx).split( - 1, dim=1) + audio_start_idx).split(1, + dim=1) else: audio_llm_pos_ids_list = [] added_audio_len += min(t_ntoken_per_chunk,