piercemaloney
/

llemma_7b

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Pierce Maloney commited on Apr 14

Commit

c5184ea

•

1 Parent(s): 31cf0d3

Admitted bad words

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -30,7 +30,7 @@ class EndpointHandler():
         # 3070, 10456, [313, 334], [29898, 1068] corresponds to "(*", and we do not want to output a comment
         # 13 is a newline character
         # [1976, 441, 29889], [4920, 441, 29889] is "Abort." [4920, 18054, 29889] is "Aborted."
-        # [2087, 29885, 4430, 29889] is "Admitted."
         bad_words_ids = [[3070], [313, 334], [10456], [13], [1976, 441, 29889], [2087, 29885, 4430, 29889], [4920, 441], [4920, 441, 29889], [4920, 18054, 29889], [29898, 1068]]
         bad_words_ids.extend(additional_bad_words_ids)

         # 3070, 10456, [313, 334], [29898, 1068] corresponds to "(*", and we do not want to output a comment
         # 13 is a newline character
         # [1976, 441, 29889], [4920, 441, 29889] is "Abort." [4920, 18054, 29889] is "Aborted."
+        # [2087, 29885, 4430, 29889], [3253, 29885, 4430, 29889] is "Admitted."
         bad_words_ids = [[3070], [313, 334], [10456], [13], [1976, 441, 29889], [2087, 29885, 4430, 29889], [4920, 441], [4920, 441, 29889], [4920, 18054, 29889], [29898, 1068]]
         bad_words_ids.extend(additional_bad_words_ids)