Skip to content

Commit 8645863

Browse files
committed
fix: add tokenizer customization back
1 parent fc3034a commit 8645863

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

bigcodebench/provider/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,8 @@ def make_model(
4141
instruction_prefix=instruction_prefix,
4242
response_prefix=response_prefix,
4343
trust_remote_code=trust_remote_code,
44+
tokenizer_name=tokenizer_name,
45+
tokenizer_legacy=tokenizer_legacy,
4446
)
4547
elif backend == "hf":
4648
from bigcodebench.provider.hf import HuggingFaceDecoder
@@ -58,6 +60,8 @@ def make_model(
5860
response_prefix=response_prefix,
5961
attn_implementation=attn_implementation,
6062
trust_remote_code=trust_remote_code,
63+
tokenizer_name=tokenizer_name,
64+
tokenizer_legacy=tokenizer_legacy,
6165
)
6266
elif backend == "openai":
6367
from bigcodebench.provider.openai import OpenAIChatDecoder

0 commit comments

Comments
 (0)