James-Edmunds commited on
Commit
60bea6d
·
verified ·
1 Parent(s): fe7fbaa

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. src/generator/generator.py +2 -2
src/generator/generator.py CHANGED
@@ -293,10 +293,10 @@ User Request: {question}"""
293
  llm = ChatOpenAI(
294
  temperature=0.95,
295
  model_name=Settings.LLM_MODEL,
296
- max_tokens=2000,
297
  top_p=0.9,
298
  presence_penalty=0.25,
299
- frequency_penalty=0.2
 
300
  )
301
 
302
  # Create QA chain
 
293
  llm = ChatOpenAI(
294
  temperature=0.95,
295
  model_name=Settings.LLM_MODEL,
 
296
  top_p=0.9,
297
  presence_penalty=0.25,
298
+ frequency_penalty=0.2,
299
+ model_kwargs={"max_completion_tokens": 2000},
300
  )
301
 
302
  # Create QA chain