Jofthomas HF staff commited on
Commit
60d22a5
·
1 Parent(s): 14fa9b7

Update TextGen/router.py

Browse files
Files changed (1) hide show
  1. TextGen/router.py +1 -1
TextGen/router.py CHANGED
@@ -84,7 +84,7 @@ def generate_text(messages: List[str], npc:str):
84
  # Initialize the LLM
85
  llm = ChatGoogleGenerativeAI(
86
  model="gemini-1.5-pro-latest",
87
- max_output_tokens=300,
88
  temperature=1,
89
  safety_settings={
90
  HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
 
84
  # Initialize the LLM
85
  llm = ChatGoogleGenerativeAI(
86
  model="gemini-1.5-pro-latest",
87
+ max_output_tokens=100,
88
  temperature=1,
89
  safety_settings={
90
  HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,