samzito12 commited on
Commit
a8e01ad
·
1 Parent(s): 5558821

try to improve the inference

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ def chat(message, history):
46
  with torch.no_grad():
47
  outputs = model.generate(
48
  **inputs,
49
- max_new_tokens=128,
50
  temperature=0.7,
51
  do_sample=True,
52
  use_cache=True,
 
46
  with torch.no_grad():
47
  outputs = model.generate(
48
  **inputs,
49
+ max_new_tokens=256,
50
  temperature=0.7,
51
  do_sample=True,
52
  use_cache=True,