Avoid negative values

Signed-off-by: DarkLight1337 <[email protected]>
vllm-project · Jan 3, 2025 · fb8a2e1 · fb8a2e1
1 parent b29f6ff
commit fb8a2e1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/vllm/model_executor/models/llava_onevision.py b/vllm/model_executor/models/llava_onevision.py
@@ -192,7 +192,7 @@ def _get_max_video_frames(
         max_total_tokens = self.ctx.model_config.max_model_len
         max_total_frames = int(max_total_tokens / self._get_max_frame_tokens())
 
-        return (max_total_frames - num_images * i2f) // max(num_videos, 1)
+        return max(max_total_frames - num_images * i2f, 0) // max(num_videos, 1)
 
     def _get_max_video_tokens(self) -> int:
         return self._get_max_frame_tokens() * self._get_max_video_frames()

diff --git a/vllm/model_executor/models/qwen2_vl.py b/vllm/model_executor/models/qwen2_vl.py
@@ -780,7 +780,7 @@ def _get_max_video_frames(
 
         max_total_tokens = self.ctx.model_config.max_model_len
         max_total_frames = int(max_total_tokens / self._get_max_image_tokens())
-        return (max_total_frames - num_images * i2f) // max(num_videos, 1)
+        return max(max_total_frames - num_images * i2f, 0) // max(num_videos, 1)
 
     def _get_max_video_tokens(self) -> int:
         return self._get_max_image_tokens() * self._get_max_video_frames()