Skip to content

Commit

Permalink
Updated vllm imports in vllm_causallms.py (EleutherAI#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 authored and notrichardren committed May 31, 2024
1 parent 879409a commit 53eeef9
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 53eeef9

Please sign in to comment.