Skip to content

GitLab

Commit 06eb4e1b authored by lintangsutawika's avatar lintangsutawika
Browse files

changed origin of get_model

parent c75ce009
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
lm_eval/evaluator.py
View file @ 06eb4e1b
...@@ -10,6 +10,7 @@ import lm_eval.api ...@@ -10,6 +10,7 @@ import lm_eval.api
import lm_eval.tasks import lm_eval.tasks
import lm_eval.models import lm_eval.models
import lm_eval.api.metrics import lm_eval.api.metrics
import lm_eval.api.registry
from lm_eval.utils import ( from lm_eval.utils import (
positional_deprecated, positional_deprecated,
...@@ -71,7 +72,7 @@ def simple_evaluate( ...@@ -71,7 +72,7 @@ def simple_evaluate(
if isinstance(model, str): if isinstance(model, str):
if model_args is None: if model_args is None:
model_args = "" model_args = ""
lm = lm_eval.api.model.get_model(model).create_from_arg_string( lm = lm_eval.api.registry.get_model(model).create_from_arg_string(
model_args, {"batch_size": batch_size, "device": device} model_args, {"batch_size": batch_size, "device": device}
) )
else: else:
......
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