wangzhang commited on
Commit
40ea808
·
1 Parent(s): 6ba060e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -39,10 +39,10 @@ def generate(
39
  chat_history: list[tuple[str, str]],
40
  system_prompt: str,
41
  max_new_tokens: int = 1024,
42
- temperature: float = 0.1,
43
- top_p: float = 0.99,
44
- top_k: int = 1,
45
- repetition_penalty: float = 1.0,
46
  ) -> Iterator[str]:
47
  conversation = []
48
  if system_prompt:
 
39
  chat_history: list[tuple[str, str]],
40
  system_prompt: str,
41
  max_new_tokens: int = 1024,
42
+ temperature: float = 0.6,
43
+ top_p: float = 0.9,
44
+ top_k: int = 50,
45
+ repetition_penalty: float = 1.2,
46
  ) -> Iterator[str]:
47
  conversation = []
48
  if system_prompt: