[Fix] Multi-images loading error (#1218)
This commit is contained in:
committed by
GitHub
parent
7514b9f8d3
commit
3579162ab1
@@ -428,7 +428,7 @@ class TokenizerManager:
|
||||
|
||||
async def _get_pixel_values(self, image_data):
|
||||
if isinstance(image_data, list) and len(image_data) > 0:
|
||||
return await self._get_pixel_values_internal(image_data[0])
|
||||
return await self._get_pixel_values_internal(image_data)
|
||||
elif isinstance(image_data, str):
|
||||
return await self._get_pixel_values_internal(image_data)
|
||||
else:
|
||||
|
||||
@@ -114,8 +114,8 @@ class TestOpenAIVisionServer(unittest.TestCase):
|
||||
text = response.choices[0].message.content
|
||||
assert isinstance(text, str)
|
||||
print(text)
|
||||
assert "man" in text or "cab" in text, text
|
||||
# assert "logo" in text, text
|
||||
assert "man" in text and "taxi" in text, text
|
||||
assert "logo" in text, text
|
||||
assert response.id
|
||||
assert response.created
|
||||
assert response.usage.prompt_tokens > 0
|
||||
|
||||
Reference in New Issue
Block a user