Skip to content

GitLab

Commit 1f504aa4 authored by baberabb's avatar baberabb
Browse files

remove unused imports

parent 694dc642
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 4 deletions
lm_eval/models/vllm_causallms.py
View file @ 1f504aa4
from collections import defaultdict from collections import defaultdict
import os
from itertools import islice
from typing import List, Tuple, Optional, Literal, Union, Any from typing import List, Tuple, Optional, Literal, Union, Any
from transformers import AutoTokenizer from transformers import AutoTokenizer
from lm_eval.api.instance import Instance from lm_eval.api.instance import Instance
...@@ -10,8 +8,6 @@ from tqdm import tqdm ...@@ -10,8 +8,6 @@ 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 from ray.util.multiprocessing import Pool
import multiprocessing
try: try:
from vllm import LLM, SamplingParams from vllm import LLM, SamplingParams
......
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