Skip to content

GitLab

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

fix imports

parent 055d99b5
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
lm_eval/models/vllm_causallms.py
View file @ 6359f083
...@@ -7,10 +7,10 @@ import copy ...@@ -7,10 +7,10 @@ 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 ray.util.multiprocessing import Pool
try: try:
from vllm import LLM, SamplingParams from vllm import LLM, SamplingParams
from ray.util.multiprocessing import Pool
except ModuleNotFoundError: except ModuleNotFoundError:
pass 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