Commit 6359f083 authored by baberabb's avatar baberabb
Browse files

fix imports

parent 055d99b5
......@@ -7,10 +7,10 @@ import copy
from tqdm import tqdm
from lm_eval.api.registry import register_model
from lm_eval import utils
from ray.util.multiprocessing import Pool
try:
from vllm import LLM, SamplingParams
from ray.util.multiprocessing import Pool
except ModuleNotFoundError:
pass
......
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