howard-hou commited on
Commit
bcbb771
1 Parent(s): 7fa18f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -48,7 +48,7 @@ def generate(
48
  alpha_frequency = countPenalty,
49
  alpha_presence = presencePenalty,
50
  token_ban = [], # ban the generation of some tokens
51
- token_stop = [0]) # stop generation whenever you see any token here
52
  ctx = ctx.strip()
53
  all_tokens = []
54
  out_last = 0
 
48
  alpha_frequency = countPenalty,
49
  alpha_presence = presencePenalty,
50
  token_ban = [], # ban the generation of some tokens
51
+ token_stop = [0, 261]) # stop generation whenever you see any token here
52
  ctx = ctx.strip()
53
  all_tokens = []
54
  out_last = 0