Update handler.py
Browse files- handler.py +2 -2
handler.py
CHANGED
@@ -22,8 +22,8 @@ class EndpointHandler:
|
|
22 |
"""
|
23 |
allowed_ids = set()
|
24 |
for word in vocab_list:
|
25 |
-
# Generate all variations: plain, space-prefixed,
|
26 |
-
variations = {word, " " + word, word.capitalize(), " " + word.capitalize()
|
27 |
|
28 |
# Add token IDs for all variations
|
29 |
for variation in variations:
|
|
|
22 |
"""
|
23 |
allowed_ids = set()
|
24 |
for word in vocab_list:
|
25 |
+
# Generate all variations: plain, space-prefixed, and capitalized
|
26 |
+
variations = {word, " " + word, word.capitalize(), " " + word.capitalize()}
|
27 |
|
28 |
# Add token IDs for all variations
|
29 |
for variation in variations:
|