fix split ratio for VLLM backend

rhymes-ai · Oct 4, 2024 · 19f2ca9 · 19f2ca9
1 parent 279ddbd
commit 19f2ca9
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/aria/vllm/aria.py b/aria/vllm/aria.py
@@ -902,14 +902,25 @@ def split_image(
     image: Image.Image,
     split_image: bool,
     split_ratio: List[List[int]] = [
-        [1, 1],
         [1, 2],
         [1, 3],
         [1, 4],
+        [1, 5],
+        [1, 6],
+        [1, 7],
+        [1, 8],
+        [2, 4],
+        [2, 3],
         [2, 2],
         [2, 1],
         [3, 1],
+        [3, 2],
         [4, 1],
+        [4, 2],
+        [5, 1],
+        [6, 1],
+        [7, 1],
+        [8, 1],
     ],
     patch_size: int = 980,
 ) -> List[Image.Image]: