fix: image size uneven across unit tests

libs/infinity_emb/infinity_emb/transformer/abstract.py

@@ -120,7 +120,7 @@ def encode_post(
 
     def warmup(self, *, batch_size: int = 64, n_tokens=1) -> tuple[float, float, str]:
         sample_text = ["warm " * n_tokens] * max(1, batch_size // 2)
-        sample_image = [Image.new("RGB", (100, 1000), (255, 255, 255))] * max(1, batch_size // 2)  # type: ignore
+        sample_image = [Image.new("RGB", (128, 128), (255, 255, 255))] * max(1, batch_size // 2)  # type: ignore
         inp = [
             # TODO: warmup for images
             ImageInner(content=ImageSingle(image=img), future=None)  # type: ignore

libs/infinity_emb/infinity_emb/transformer/vision/torch_vision.py

@@ -45,7 +45,7 @@ def __init__(self, *, engine_args: "EngineArgs"):
         )
         config = AutoConfig.from_pretrained(**base_config)
         self.is_colipali = config.architectures[0] in IMAGE_COL_MODELS
-        self.mock_image = Image.new("RGB", (16, 16), color="black")
+        self.mock_image = Image.new("RGB", (128, 128), color="black")
 
         extra_model_args = dict(**base_config)
         extra_processor_args = dict(**base_config)

libs/infinity_emb/tests/unit_test/transformer/vision/test_torch_vision.py

@@ -82,8 +82,8 @@ def test_colpali(dtype, image_sample):
         images = [
             image,
             image,
-            Image.new("RGB", (32, 32), color="white"),
-            Image.new("RGB", (16, 16), color="black"),
+            Image.new("RGB", (128, 128), color="white"),
+            Image.new("RGB", (128, 128), color="black"),
         ]
         queries = [
             "a photo of a cat",