Spaces:
Running
on
Zero
Running
on
Zero
Rijgersberg
commited on
Commit
•
54c6ef2
1
Parent(s):
d0af199
Set generation parameters
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 = 0
|
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 = 1.0,
|
37 |
top_p: float = 0.9,
|
38 |
+
top_k: int = 10,
|
39 |
repetition_penalty: float = 1.2,
|
40 |
) -> Iterator[str]:
|
41 |
conversation = []
|