0.38.2: fix multi-image for ovis1.6

matatonic · Oct 9, 2024 · 426e6c5 · 426e6c5
1 parent c092d15
commit 426e6c5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/backend/ovis16.py b/backend/ovis16.py
@@ -30,13 +30,14 @@ def __init__(self, model_id: str, device: str, device_map: str = 'auto', extra_p
     async def stream_chat_with_images(self, request: ImageChatRequest) -> AsyncGenerator[str, None]:
         conversation = []
         images = []
+
         for m in request.messages:
             content = ''
             for c in m.content:
                 if c.type == 'image_url':
                     image = await url_to_image(c.image_url.url)
                     images.extend([image])
-                    content = IMAGE_TOKEN + '\n' + content
+                    content += IMAGE_TOKEN + '\n'
                 elif c.type == 'text':
                     content += c.text