Update README.md
Browse files
README.md
CHANGED
|
@@ -163,7 +163,7 @@ def uniformisation(s):
|
|
| 163 |
def chat(
|
| 164 |
query: str,
|
| 165 |
history: Optional[List[Dict]] = None,
|
| 166 |
-
temperature: float = 0.
|
| 167 |
top_p: float = 1.0,
|
| 168 |
top_k: float = 0,
|
| 169 |
repetition_penalty: float = 1.2,
|
|
@@ -387,7 +387,7 @@ def stream(message, history):
|
|
| 387 |
**inputs,
|
| 388 |
max_new_tokens=1,
|
| 389 |
do_sample=True,
|
| 390 |
-
temperature=0.
|
| 391 |
pad_token_id=tokenizer.pad_token_id
|
| 392 |
)
|
| 393 |
|
|
|
|
| 163 |
def chat(
|
| 164 |
query: str,
|
| 165 |
history: Optional[List[Dict]] = None,
|
| 166 |
+
temperature: float = 0.85,
|
| 167 |
top_p: float = 1.0,
|
| 168 |
top_k: float = 0,
|
| 169 |
repetition_penalty: float = 1.2,
|
|
|
|
| 387 |
**inputs,
|
| 388 |
max_new_tokens=1,
|
| 389 |
do_sample=True,
|
| 390 |
+
temperature=0.8,
|
| 391 |
pad_token_id=tokenizer.pad_token_id
|
| 392 |
)
|
| 393 |
|