We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f6fee3a commit 2073a29Copy full SHA for 2073a29
src/lighteval/models/litellm_model.py
@@ -255,12 +255,18 @@ def greedy_until(
255
def tokenizer(self):
256
return self._tokenizer
257
258
+ def _encode(self, text: str):
259
+ enc = encode(model=self.model, text=text)
260
+ if hasattr(enc, "ids"):
261
+ return enc.ids
262
+ return enc
263
+
264
def tok_encode(self, text: str | list[str]):
265
if isinstance(text, list):
- toks = [encode(model=self.model, text=t["content"]) for t in text]
266
+ toks = [self._encode(t["content"]) for t in text]
267
toks = [tok for tok in toks if tok]
268
return toks
- return encode(model=self.model, text=text)
269
+ return self._encode(text)
270
271
@property
272
def add_special_tokens(self) -> bool:
0 commit comments