Quality fixes

neuralmagic · Nov 5, 2024 · 570670b · 570670b
1 parent a441dad
commit 570670b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/guidellm/backend/openai.py b/src/guidellm/backend/openai.py
@@ -179,7 +179,7 @@ def _build_messages(self, request: TextGenerationRequest) -> Dict:
                 stream = io.BytesIO()
                 im_format = image.image.format or "PNG"
                 image.image.save(stream, format=im_format)
-                im_b64 = base64.b64encode(stream.getvalue()).decode("ascii")
+                im_b64 = base64.b64encode(stream.getvalue()).decode("utf-8")
                 image_url = {"url": f"data:image/{im_format.lower()};base64,{im_b64}"}
                 content.append({"type": "image_url", "image_url": image_url})
 

diff --git a/src/guidellm/utils/images.py b/src/guidellm/utils/images.py
@@ -65,4 +65,4 @@ def load_images(data: str) -> List[ImageDescriptor]:
                     )
                 )
 
-    return images
+        return images