NCTCMumbai commited on
Commit
a1224db
1 Parent(s): b4c0def

Update backend/query_llm.py

Browse files
Files changed (1) hide show
  1. backend/query_llm.py +2 -2
backend/query_llm.py CHANGED
@@ -11,8 +11,8 @@ from transformers import AutoTokenizer
11
 
12
  #tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1")
13
  tokenizer = AutoTokenizer.from_pretrained("mistralai/Mixtral-8x7B-Instruct-v0.1")
14
- temperature = 0.4
15
- top_p = 0.6
16
  repetition_penalty = 1.2
17
 
18
  OPENAI_KEY = getenv("OPENAI_API_KEY")
 
11
 
12
  #tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1")
13
  tokenizer = AutoTokenizer.from_pretrained("mistralai/Mixtral-8x7B-Instruct-v0.1")
14
+ temperature = 0.5
15
+ top_p = 0.7
16
  repetition_penalty = 1.2
17
 
18
  OPENAI_KEY = getenv("OPENAI_API_KEY")