Fix support to images

Author: anmarques

src/guidellm/core/request.py

@@ -1,5 +1,5 @@
 import uuid
-from typing import Any, Dict, List, Optional, Tuple
+from typing import Any, Dict, List, Optional
 
 from pydantic import Field
 
@@ -42,7 +42,7 @@ def number_images(self) -> int:
             return len(self.images)
 
     @property
-    def image_resolution(self) -> Tuple[int]:
+    def image_resolution(self) -> List[int]:
         if self.images is None:
             return None
         else:

src/guidellm/request/emulated.py

@@ -31,7 +31,8 @@ class EmulatedConfig:
         generated_tokens_min (Optional[int]): Minimum number of generated tokens.
         generated_tokens_max (Optional[int]): Maximum number of generated tokens.
         images (Optional[int]): Number of images.
-        image_resultion (Optional[List[int]]): Resolution of images.
+        width (Optional[int]): Width of images.
+        height (Optional[int]): Height of images.
     """
 
     @staticmethod
@@ -108,7 +109,8 @@ def create_config(config: Optional[Union[str, Path, Dict]]) -> "EmulatedConfig":
     generated_tokens_max: Optional[int] = None
 
     images: int = 0
-    image_resolution = None
+    width: int = None
+    height: int = None
 
     def __post_init__(self):
         if self.images is not None and self.image_resultion is not None and self.images > 0:
@@ -337,7 +339,7 @@ def __init__(
             settings.emulated_data.filter_end,
         )
         if self._config.images > 0:
-            self._images = load_images(settings.emulated_data.image_source, self._config.image_resolution)
+            self._images = load_images(settings.emulated_data.image_source, [self._config.width, self._config.height])
         self._rng = np.random.default_rng(random_seed)
 
         # NOTE: Must be after all the parameters since the queue population

src/guidellm/utils/images.py

@@ -6,7 +6,7 @@
 from bs4 import BeautifulSoup
 from loguru import logger
 from PIL import Image
-from pydantic import ConfigDict, Field
+from pydantic import ConfigDict, Field, computed_field
 
 from guidellm.config import settings
 from guidellm.core.serializable import Serializable
@@ -26,6 +26,14 @@ class ImageDescriptor(Serializable):
         description="Image filename.",
     )
 
+    @computed_field # type: ignore[misc]
+    @property
+    def image_resolution(self) -> List[int]:
+        if self.images is None:
+            return None
+        else:
+            return [im.size for im in self.images]
+
 
 def load_images(data: str, image_resolution: Optional[List[int]]) -> List[ImageDescriptor]:
     """