Detsutut commited on
Commit
80797a0
1 Parent(s): 7ebea95

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -49,7 +49,8 @@ def generate_text(input_text, max_new_tokens=512, temperature=1, system_prompt="
49
  max_new_tokens=max_new_tokens,
50
  temperature=temperature,
51
  return_full_text = False,
52
- forced_eos_token_id=tokenizer.encode("[...]")[1]
 
53
  )
54
  generated_text = output[0]['generated_text']
55
 
 
49
  max_new_tokens=max_new_tokens,
50
  temperature=temperature,
51
  return_full_text = False,
52
+ forced_eos_token_id=tokenizer.encode("[...]")[1],
53
+ pad_token_id=tokenizer.eos_token_id
54
  )
55
  generated_text = output[0]['generated_text']
56