Merge pull request #4234 from kimdwkimdw/patch-1

Support vllm==0.5.0
This commit is contained in:
hoshi-hiyouga 2024-06-12 16:39:09 +08:00 committed by GitHub
commit 656b2bbdaf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions

View File

@ -13,7 +13,10 @@ from .base_engine import BaseEngine, Response
if is_vllm_available():
from vllm import AsyncEngineArgs, AsyncLLMEngine, RequestOutput, SamplingParams
from vllm.lora.request import LoRARequest
from vllm.sequence import MultiModalData
try:
from vllm.multimodal import MultiModalData # vllm==0.5.0
except ImportError:
from vllm.sequence import MultiModalData # vllm<0.5.0
if TYPE_CHECKING: