PraneetNS commited on
Commit
ced27a1
·
verified ·
1 Parent(s): 7ae37cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -110,8 +110,8 @@ def generate(message, history):
110
  with torch.no_grad():
111
  outputs = model.generate(
112
  **inputs,
113
- max_new_tokens=512,
114
- temperature=0.2,
115
  top_p=0.95,
116
  do_sample=True,
117
  repetition_penalty=1.1,
 
110
  with torch.no_grad():
111
  outputs = model.generate(
112
  **inputs,
113
+ max_new_tokens=1024,
114
+ temperature=0.3,
115
  top_p=0.95,
116
  do_sample=True,
117
  repetition_penalty=1.1,