We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9b87874 commit d46a99cCopy full SHA for d46a99c
lm_eval/utils.py
@@ -162,7 +162,7 @@ def parse_infill(code, tokenizer):
162
prefix, rest = code.split("<fim-suffix>", 1)
163
suffix, infill = rest.split("<fim-middle>", 1)
164
infill = infill.split("<|endoftext|>")[0]
165
- elif model_id in ["bigcode/large-model", "bigcode/temp-model"]:
+ elif model_id in ["bigcode/starcoder", "bigcode/starcoderbase"]:
166
prefix, rest = code.split("<fim_suffix>", 1)
167
suffix, infill = rest.split("<fim_middle>", 1)
168
0 commit comments