Skip to content

GitLab

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

switch to using get_model()

parent d362dfee
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
lm_eval/evaluator.py
View file @ 4221d30f
...@@ -77,7 +77,7 @@ def simple_evaluate( ...@@ -77,7 +77,7 @@ def simple_evaluate(
model_args, {"batch_size": batch_size, "max_batch_size": max_batch_size, "device": device} model_args, {"batch_size": batch_size, "max_batch_size": max_batch_size, "device": device}
) )
elif isinstance(model, transformers.PreTrainedModel): elif isinstance(model, transformers.PreTrainedModel):
lm = HFLM( lm = lm_eval.models.get_model("hf-causal")(
pretrained=model, pretrained=model,
batch_size=batch_size, 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