From 7f6d267e989a7186da76ef275e110465cf1acc88 Mon Sep 17 00:00:00 2001
From: marko1616 <45327989+marko1616@users.noreply.github.com>
Date: Thu, 22 Aug 2024 12:14:34 +0800
Subject: [PATCH] Update chat.py

Former-commit-id: a4f1de9d82743bfd35a6dbc1027242533b22ed5f
---
 src/llamafactory/api/chat.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/llamafactory/api/chat.py b/src/llamafactory/api/chat.py
index 72b2ae50..1010aab1 100644
--- a/src/llamafactory/api/chat.py
+++ b/src/llamafactory/api/chat.py
@@ -17,6 +17,7 @@ import io
 import json
 import os
 import uuid
+import numpy
 from typing import TYPE_CHECKING, AsyncGenerator, Dict, List, Optional, Tuple
 
 from ..data import Role as DataRole
@@ -112,7 +113,7 @@ def _process_request(
                     else:  # web uri
                         image_path = requests.get(image_url, stream=True).raw
 
-                    image = Image.open(image_path).convert("RGB")
+                    image = numpy.array(Image.open(image_path).convert("RGB"))
         else:
             input_messages.append({"role": ROLE_MAPPING[message.role], "content": message.content})