Update vllm/multimodal/image.py

Co-authored-by: Cyrus Leung <cyrus.tl.leung@gmail.com>
vllm-project · Aug 21, 2024 · 7c8f82f · 7c8f82f
1 parent b4a0bcc
commit 7c8f82f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vllm/multimodal/image.py b/vllm/multimodal/image.py
@@ -77,11 +77,11 @@ def repeat_and_pad_image_tokens(
         prompt_parts = prompt.split(image_token_str,
                                     maxsplit=len(repeat_count))
         new_prompt = ""
-        for i in range(len(repeat_count)):
+        for repeat_count_item in repeat_count:
             replacement_str = "".join(
                 repeat_and_pad_token(
                     image_token_str,
-                    repeat_count=repeat_count[i],
+                    repeat_count=repeat_count_item,
                     pad_token_left=pad_token_str_left,
                     pad_token_right=pad_token_str_right,
                 ))