Update processing_videollama3.py

processing_videollama3.py

@@ -681,14 +681,15 @@ class Videollama3Qwen2Processor(ProcessorMixin):
         kwargs.pop("padding")
         kwargs.pop("padding_side")
 
-        image_idx = 0
-        while DEFAULT_IMAGE_TOKEN in text:
-            num_tokens = self._get_visual_seq_len(grid_sizes[image_idx])
-            text = text.replace(DEFAULT_IMAGE_TOKEN, "<placeholder>" * num_tokens, 1)
-            image_idx += 1
-        text = text.replace("<placeholder>", DEFAULT_IMAGE_TOKEN)
- 
-        assert len(grid_sizes) == image_idx, "Number of images does not match the number of image tokens in the text."
+        if len(grid_sizes) > 0:
+            image_idx = 0
+            while DEFAULT_IMAGE_TOKEN in text:
+                num_tokens = self._get_visual_seq_len(grid_sizes[image_idx])
+                text = text.replace(DEFAULT_IMAGE_TOKEN, "<placeholder>" * num_tokens, 1)
+                image_idx += 1
+            text = text.replace("<placeholder>", DEFAULT_IMAGE_TOKEN)
+    
+            assert len(grid_sizes) == image_idx, "Number of images does not match the number of image tokens in the text."
 
         text_inputs = self.tokenizer(text, **kwargs)
         return text_inputs