"vscode:/vscode.git/clone" did not exist on "a9c1e3a31dd7025e3e4dab9003f091424706d770"
Commit 4221d30f authored by haileyschoelkopf's avatar haileyschoelkopf
Browse files

switch to using get_model()

parent d362dfee
......@@ -77,7 +77,7 @@ def simple_evaluate(
model_args, {"batch_size": batch_size, "max_batch_size": max_batch_size, "device": device}
)
elif isinstance(model, transformers.PreTrainedModel):
lm = HFLM(
lm = lm_eval.models.get_model("hf-causal")(
pretrained=model,
batch_size=batch_size,
)
......
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