LymphSteamer commited on
Commit
b7bf584
·
verified ·
1 Parent(s): 2383d74

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ def haiku_generate():
20
  with torch.no_grad():
21
  outputs = model.generate(
22
  **inputs,
23
- max_new_tokens=25,
24
  eos_token_id=tokenizer.eos_token_id, # 終わりを明示!
25
  pad_token_id=tokenizer.pad_token_id,
26
  do_sample=True,
 
20
  with torch.no_grad():
21
  outputs = model.generate(
22
  **inputs,
23
+ max_new_tokens=20,
24
  eos_token_id=tokenizer.eos_token_id, # 終わりを明示!
25
  pad_token_id=tokenizer.pad_token_id,
26
  do_sample=True,