mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-01-20 16:54:33 +08:00
Signed-off-by: AzizCode92 <azizbenothman76@gmail.com> Signed-off-by: Aziz <azizbenothman76@gmail.com> Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> Co-authored-by: Cyrus Leung <tlleungac@connect.ust.hk>
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
|
|
from .interfaces import (HasInnerState, SupportsLoRA, SupportsMRoPE,
|
|
SupportsMultiModal, SupportsPP, SupportsTranscription,
|
|
SupportsV0Only, has_inner_state, supports_lora,
|
|
supports_mrope, supports_multimodal, supports_pp,
|
|
supports_transcription, supports_v0_only)
|
|
from .interfaces_base import (VllmModelForPooling, VllmModelForTextGeneration,
|
|
is_pooling_model, is_text_generation_model)
|
|
from .registry import ModelRegistry
|
|
|
|
__all__ = [
|
|
"ModelRegistry",
|
|
"VllmModelForPooling",
|
|
"is_pooling_model",
|
|
"VllmModelForTextGeneration",
|
|
"is_text_generation_model",
|
|
"HasInnerState",
|
|
"has_inner_state",
|
|
"SupportsLoRA",
|
|
"supports_lora",
|
|
"SupportsMultiModal",
|
|
"supports_multimodal",
|
|
"SupportsMRoPE",
|
|
"supports_mrope",
|
|
"SupportsPP",
|
|
"supports_pp",
|
|
"SupportsTranscription",
|
|
"supports_transcription",
|
|
"SupportsV0Only",
|
|
"supports_v0_only",
|
|
]
|