CreitinGameplays commited on
Commit
4fbfff8
·
verified ·
1 Parent(s): 63b5b08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -28,8 +28,8 @@ if torch.cuda.is_available():
28
  def generate(
29
  message: str,
30
  chat_history: list[tuple[str, str]],
31
- max_new_tokens: int = 1024,
32
- temperature: float = 0.6,
33
  top_p: float = 0.9,
34
  top_k: int = 50,
35
  repetition_penalty: float = 1.2,
 
28
  def generate(
29
  message: str,
30
  chat_history: list[tuple[str, str]],
31
+ max_new_tokens: int = 512,
32
+ temperature: float = 0.2,
33
  top_p: float = 0.9,
34
  top_k: int = 50,
35
  repetition_penalty: float = 1.2,