eneSadi commited on
Commit
1545115
·
unverified ·
1 Parent(s): 629dd02

max new token update

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ async def ask(request: Request):
49
  print("Tokenizer process completed")
50
 
51
  print("Model process started")
52
- outputs = model.generate(**input_ids, max_new_tokens=256)
53
 
54
  print("Tokenizer decode process started")
55
  answer = tokenizer.decode(outputs[0]).split("<end_of_turn>")[1].strip()
 
49
  print("Tokenizer process completed")
50
 
51
  print("Model process started")
52
+ outputs = model.generate(**input_ids, max_new_tokens=512)
53
 
54
  print("Tokenizer decode process started")
55
  answer = tokenizer.decode(outputs[0]).split("<end_of_turn>")[1].strip()