Spaces:
Sleeping
Sleeping
Update chat_client.py
Browse files- chat_client.py +1 -1
chat_client.py
CHANGED
|
@@ -17,7 +17,7 @@ def format_prompt(message, history):
|
|
| 17 |
return prompt
|
| 18 |
|
| 19 |
def chat(
|
| 20 |
-
prompt, history, chat_client = "mistralai/Mistral-7B-Instruct-v0.1",temperature=0.9, max_new_tokens=
|
| 21 |
):
|
| 22 |
client = InferenceClient(
|
| 23 |
chat_client,
|
|
|
|
| 17 |
return prompt
|
| 18 |
|
| 19 |
def chat(
|
| 20 |
+
prompt, history, chat_client = "mistralai/Mistral-7B-Instruct-v0.1",temperature=0.9, max_new_tokens=2048, top_p=0.95, repetition_penalty=1.0,
|
| 21 |
):
|
| 22 |
client = InferenceClient(
|
| 23 |
chat_client,
|