max tokens for multimodal calls

zou-group · Jul 5, 2024 · 8318dee · 8318dee
1 parent e9c1240
commit 8318dee
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/textgrad/engine/anthropic.py b/textgrad/engine/anthropic.py
@@ -103,7 +103,7 @@ def _format_content(self, content: List[Union[str, bytes]]) -> List[dict]:
         return formatted_content
 
     def _generate_multimodal(
-        self, content: List[Union[str, bytes]], system_prompt=None, temperature=0, max_tokens=300, top_p=0.99
+        self, content: List[Union[str, bytes]], system_prompt=None, temperature=0, max_tokens=2000, top_p=0.99
     ):
         sys_prompt_arg = system_prompt if system_prompt else self.system_prompt
         formatted_content = self._format_content(content)

diff --git a/textgrad/engine/openai.py b/textgrad/engine/openai.py
@@ -109,7 +109,7 @@ def _format_content(self, content: List[Union[str, bytes]]) -> List[dict]:
         return formatted_content
 
     def _generate_multimodal(
-        self, content: List[Union[str, bytes]], system_prompt=None, temperature=0, max_tokens=300, top_p=0.99
+        self, content: List[Union[str, bytes]], system_prompt=None, temperature=0, max_tokens=2000, top_p=0.99
     ):
         sys_prompt_arg = system_prompt if system_prompt else self.system_prompt
         formatted_content = self._format_content(content)