From 8d779c6cbfaf46b3bd9f8993bcde3b6b430492d2 Mon Sep 17 00:00:00 2001 From: Yazhou Cao Date: Mon, 19 Feb 2024 22:49:38 -0800 Subject: [PATCH] Fix ubuntu-specific type error --- lmm_tools/lmm/lmm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lmm_tools/lmm/lmm.py b/lmm_tools/lmm/lmm.py index c2efde5d..e6f954a4 100644 --- a/lmm_tools/lmm/lmm.py +++ b/lmm_tools/lmm/lmm.py @@ -58,7 +58,7 @@ def generate(self, prompt: str, image: Optional[Union[str, Path]]) -> str: ) response = self.client.chat.completions.create( - model="gpt-4-vision-preview", message=message + model="gpt-4-vision-preview", messages=message # type: ignore ) return cast(str, response.choices[0].message.content)