Update app.py
Browse files
app.py
CHANGED
@@ -46,8 +46,7 @@ def generate(
|
|
46 |
chat_history: list,
|
47 |
system_prompt: str = DANY_SYSTEM_PROMPT,
|
48 |
max_new_tokens: int = DEFAULT_MAX_NEW_TOKENS,
|
49 |
-
temperature: float = 0
|
50 |
-
top_p: float = 0.9,
|
51 |
top_k: int = 50,
|
52 |
repetition_penalty: float = 1.0,
|
53 |
) -> Iterator[str]:
|
|
|
46 |
chat_history: list,
|
47 |
system_prompt: str = DANY_SYSTEM_PROMPT,
|
48 |
max_new_tokens: int = DEFAULT_MAX_NEW_TOKENS,
|
49 |
+
temperature: float = 1.0,
|
|
|
50 |
top_k: int = 50,
|
51 |
repetition_penalty: float = 1.0,
|
52 |
) -> Iterator[str]:
|