Skip to content

Commit

Permalink
Updated vllm imports in vllm_causallms.py (#1890)
Browse files Browse the repository at this point in the history
* Reorder vllm imports in vllm_causallms.py

* Update vllm_causallms.py
  • Loading branch information
mgoin committed May 28, 2024
1 parent 0ff6ab9 commit b4cd85d
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions lm_eval/models/vllm_causallms.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,9 +21,7 @@
try:
import ray
from vllm import LLM, SamplingParams

if parse_version(version("vllm")) > parse_version("0.3.0"):
from vllm.lora.request import LoRARequest
from vllm.lora.request import LoRARequest
from vllm.transformers_utils.tokenizer import get_tokenizer
except ModuleNotFoundError:
pass
Expand Down

0 comments on commit b4cd85d

Please sign in to comment.