Update app.py
Browse files
app.py
CHANGED
@@ -10,8 +10,8 @@ llm_zephyr_7b_beta = HuggingFaceHub(
|
|
10 |
repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1",
|
11 |
task="text-generation",
|
12 |
model_kwargs={
|
13 |
-
"max_new_tokens":
|
14 |
-
"top_k":
|
15 |
"temperature": 0.1,
|
16 |
"repetition_penalty": 1.03,
|
17 |
},
|
|
|
10 |
repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1",
|
11 |
task="text-generation",
|
12 |
model_kwargs={
|
13 |
+
"max_new_tokens": 1000,
|
14 |
+
"top_k": 2,
|
15 |
"temperature": 0.1,
|
16 |
"repetition_penalty": 1.03,
|
17 |
},
|