taylorj94 commited on
Commit
127bf34
·
verified ·
1 Parent(s): c0e19b5

Update handler.py

Browse files
Files changed (1) hide show
  1. 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, capitalized, and uppercase
26
- variations = {word, " " + word, word.capitalize(), " " + word.capitalize(), word.upper(), " " + word.upper()}
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: