reab5555 commited on
Commit
083c4ca
·
verified ·
1 Parent(s): 52462b6

Update llm_loader.py

Browse files
Files changed (1) hide show
  1. llm_loader.py +3 -3
llm_loader.py CHANGED
@@ -4,7 +4,7 @@ def load_model(openai_api_key):
4
  return ChatOpenAI(
5
  model_name="gpt-4o-mini",
6
  openai_api_key=openai_api_key,
7
- temperature=0.1,
8
- max_tokens=10000,
9
- top_p=0.95
10
  )
 
4
  return ChatOpenAI(
5
  model_name="gpt-4o-mini",
6
  openai_api_key=openai_api_key,
7
+ temperature=0.01,
8
+ max_tokens=4096,
9
+ top_p=0.8
10
  )