Spaces:
Sleeping
Sleeping
howard-hou
commited on
Commit
•
bcbb771
1
Parent(s):
7fa18f3
Update app.py
Browse files
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
|