Unverified Commit 94cde109 authored by Liangsheng Yin's avatar Liangsheng Yin Committed by GitHub
Browse files

Llama3.2 vision model support (#1551)

parent 00611286
......@@ -171,7 +171,7 @@ class TestOpenAIVisionServer(unittest.TestCase):
assert isinstance(text, str)
print(text)
assert "man" in text or "cab" in text, text
assert "logo" in text, text
assert "logo" in text or '"S"' in text or "SG" in text, text
assert response.id
assert response.created
assert response.usage.prompt_tokens > 0
......@@ -363,5 +363,27 @@ class TestQWen2VLServer(TestOpenAIVisionServer):
cls.base_url += "/v1"
class TestMllamaServer(TestOpenAIVisionServer):
@classmethod
def setUpClass(cls):
cls.model = "meta-llama/Llama-3.2-11B-Vision-Instruct"
cls.base_url = DEFAULT_URL_FOR_TEST
cls.api_key = "sk-123456"
cls.process = popen_launch_server(
cls.model,
cls.base_url,
timeout=DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
api_key=cls.api_key,
other_args=[
"--chat-template",
"llama_3_vision",
],
)
cls.base_url += "/v1"
def test_video_chat_completion(self):
pass
if __name__ == "__main__":
unittest.main()
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment