Spaces:

espnet
/

SingingSDS

Sleeping

jhansss commited on Jul 10

Commit

55bdc43

1 Parent(s): 48162db

increase gemini max_output_tokens

Files changed (1) hide show

modules/llm/gemini.py CHANGED Viewed

@@ -27,7 +27,7 @@ class GeminiLLM(AbstractLLMModel):
         self,
         prompt: str,
         system_prompt: Optional[str] = None,
-        max_output_tokens: int = 1024,
         max_iterations: int = 3,
         **kwargs,
     ) -> str:

         self,
         prompt: str,
         system_prompt: Optional[str] = None,
+        max_output_tokens: int = 8192,
         max_iterations: int = 3,
         **kwargs,
     ) -> str: