NCTCMumbai
commited on
Update backend/query_llm.py
Browse files- 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.
|
15 |
-
top_p = 0.
|
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")
|