Skip to content

Commit fc3034a

Browse files
authored
fix: add trust_remote_code back
[FIX] trust_remote_code not correctly propagated to vllm/hf
2 parents e517a9e + 54794ed commit fc3034a

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

bigcodebench/provider/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@ def make_model(
4040
tp=tp,
4141
instruction_prefix=instruction_prefix,
4242
response_prefix=response_prefix,
43+
trust_remote_code=trust_remote_code,
4344
)
4445
elif backend == "hf":
4546
from bigcodebench.provider.hf import HuggingFaceDecoder
@@ -56,6 +57,7 @@ def make_model(
5657
instruction_prefix=instruction_prefix,
5758
response_prefix=response_prefix,
5859
attn_implementation=attn_implementation,
60+
trust_remote_code=trust_remote_code,
5961
)
6062
elif backend == "openai":
6163
from bigcodebench.provider.openai import OpenAIChatDecoder

0 commit comments

Comments
 (0)