Skip to content

Commit d46a99c

Browse files
benlipkinloubnabnl
andauthored
Update model names
Co-authored-by: Loubna Ben Allal <[email protected]>
1 parent 9b87874 commit d46a99c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

lm_eval/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -162,7 +162,7 @@ def parse_infill(code, tokenizer):
162162
prefix, rest = code.split("<fim-suffix>", 1)
163163
suffix, infill = rest.split("<fim-middle>", 1)
164164
infill = infill.split("<|endoftext|>")[0]
165-
elif model_id in ["bigcode/large-model", "bigcode/temp-model"]:
165+
elif model_id in ["bigcode/starcoder", "bigcode/starcoderbase"]:
166166
prefix, rest = code.split("<fim_suffix>", 1)
167167
suffix, infill = rest.split("<fim_middle>", 1)
168168
infill = infill.split("<|endoftext|>")[0]

0 commit comments

Comments
 (0)