Upload folder using huggingface_hub
Browse files
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
|