"vscode:/vscode.git/clone" did not exist on "c844ac637e44fac0441dd958ae929e3657290c9a"
Commit 6cfc0f28 authored by Baber's avatar Baber
Browse files

nit

parent 22f5ad26
......@@ -11,7 +11,7 @@ from lm_eval.models.huggingface import HFLM
class MambaLMWrapper(HFLM):
def __init__(
self,
pretrained="",
pretrained,
# To use the HF compatible variant
is_hf: bool = False,
**kwargs,
......@@ -19,6 +19,7 @@ class MambaLMWrapper(HFLM):
if "backend" in kwargs:
assert kwargs["backend"] == "causal"
self.is_hf = is_hf or (True if pretrained.endswith("hf") else False)
assert kwargs["tokenizer"] is not None, "`tokenizer` is required"
self.tokenizer = kwargs["tokenizer"]
self.pretrained = pretrained
super().__init__(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment