Rijgersberg commited on
Commit
54c6ef2
1 Parent(s): d0af199

Set generation parameters

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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.6,
37
  top_p: float = 0.9,
38
- top_k: int = 50,
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 = []