Xeltron-cloud commited on
Commit
fa9565b
·
verified ·
1 Parent(s): 05d9dc7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -44,6 +44,7 @@ async def generate_text(request: GenerateRequest):
44
  do_sample=True,
45
  early_stopping=True,
46
  repetition_penalty=1.1,
 
47
  pad_token_id=tokenizer.eos_token_id
48
  )
49
 
 
44
  do_sample=True,
45
  early_stopping=True,
46
  repetition_penalty=1.1,
47
+ eos_token_id=tokenizer.eos_token_id,
48
  pad_token_id=tokenizer.eos_token_id
49
  )
50