Spaces:
Running
Running
Commit
·
a174343
1
Parent(s):
fcdfc0f
Set temperature lower
Browse files
app.py
CHANGED
|
@@ -33,9 +33,9 @@ def generate(
|
|
| 33 |
message: str,
|
| 34 |
chat_history: list[tuple[str, str]],
|
| 35 |
max_new_tokens: int = 1024,
|
| 36 |
-
temperature: float =
|
| 37 |
top_p: float = 0.9,
|
| 38 |
-
top_k: int =
|
| 39 |
repetition_penalty: float = 1.2,
|
| 40 |
) -> Iterator[str]:
|
| 41 |
conversation = []
|
|
|
|
| 33 |
message: str,
|
| 34 |
chat_history: list[tuple[str, str]],
|
| 35 |
max_new_tokens: int = 1024,
|
| 36 |
+
temperature: float = 0.6,
|
| 37 |
top_p: float = 0.9,
|
| 38 |
+
top_k: int = 40,
|
| 39 |
repetition_penalty: float = 1.2,
|
| 40 |
) -> Iterator[str]:
|
| 41 |
conversation = []
|