Skip to content

Commit cb5cccc

Browse files
committed
fix starcoder fim constructor
1 parent 9d17482 commit cb5cccc

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

bigcode_eval/tasks/santacoder_fim.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -59,11 +59,12 @@ def __init__(
5959
fim_prefix: str = "<fim-prefix>",
6060
fim_middle: str = "<fim-middle>",
6161
fim_suffix: str = "<fim-suffix>",
62+
stop_words: List[str] = ["<|endoftext|>", "<|filename|>"],
63+
requires_execution: bool = False
6264
):
63-
stop_words = ["<|endoftext|>", "<|filename|>"]
6465
super().__init__(
6566
stop_words=stop_words,
66-
requires_execution=False,
67+
requires_execution=requires_execution,
6768
)
6869
self.fim_prefix = fim_prefix
6970
self.fim_middle = fim_middle

0 commit comments

Comments
 (0)