seko02 commited on
Commit
dd09bf7
·
verified ·
1 Parent(s): b5ae1b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ def respond(message, history):
39
  with torch.no_grad():
40
  out = model.generate(
41
  input_ids=input_ids,
42
- max_new_tokens=40,
43
  do_sample=False,
44
  temperature=0.7,
45
  eos_token_id=tokenizer.eos_token_id,
 
39
  with torch.no_grad():
40
  out = model.generate(
41
  input_ids=input_ids,
42
+ max_new_tokens=128,
43
  do_sample=False,
44
  temperature=0.7,
45
  eos_token_id=tokenizer.eos_token_id,