Commit f2890cc1 authored by baberabb's avatar baberabb
Browse files

bugfix

parent 91868979
...@@ -7,7 +7,7 @@ import copy ...@@ -7,7 +7,7 @@ import copy
from tqdm import tqdm from tqdm import tqdm
from lm_eval.api.registry import register_model from lm_eval.api.registry import register_model
from lm_eval import utils from lm_eval import utils
from vllm_causallms import LLM, SamplingParams from vllm import LLM, SamplingParams
@register_model("vllm") @register_model("vllm")
......
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