gdnartea commited on
Commit
43d116b
·
verified ·
1 Parent(s): f7e9bbd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -116,7 +116,7 @@ def generate_response(user_input):
116
  with torch.no_grad():
117
  outputs = proc_model.generate(
118
  inputs,
119
- max_new_tokens=200,
120
  )
121
 
122
  response = proc_tokenizer.batch_decode(
 
116
  with torch.no_grad():
117
  outputs = proc_model.generate(
118
  inputs,
119
+ max_new_tokens=256,
120
  )
121
 
122
  response = proc_tokenizer.batch_decode(