Skip to content

Commit 677a130

Browse files
committed
fix: rm legacy
1 parent c4ebd8d commit 677a130

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

bigcodebench/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,7 @@ def __init__(self, name: str, dataset: str, tp: int, **kwargs) -> None:
133133
if self.tokenizer_name is None:
134134
self.tokenizer_name = self.name
135135

136-
self.tokenizer = AutoTokenizer.from_pretrained(self.tokenizer_name, legacy=False, **kwargs)
136+
self.tokenizer = AutoTokenizer.from_pretrained(self.tokenizer_name, **kwargs)
137137
if self.tokenizer.chat_template is None:
138138
self.eos += extra_eos_for_direct_completion(dataset)
139139
self.llm = LLM(model=name, max_model_len=2048, **kwargs)
@@ -193,7 +193,7 @@ def __init__(self, name: str, dataset: str, **kwargs):
193193
if self.tokenizer_name is None:
194194
self.tokenizer_name = self.name
195195

196-
self.tokenizer = AutoTokenizer.from_pretrained(self.tokenizer_name, legacy=False, **kwargs)
196+
self.tokenizer = AutoTokenizer.from_pretrained(self.tokenizer_name, **kwargs)
197197

198198
if self.tokenizer.chat_template is None:
199199
self.eos += extra_eos_for_direct_completion(dataset)

0 commit comments

Comments
 (0)