31 lines
1.0 KiB
Python
31 lines
1.0 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
|
|
from .interfaces import (HasInnerState, SupportsLoRA, SupportsMultiModal,
|
|
SupportsPP, SupportsTranscription, SupportsV0Only,
|
|
has_inner_state, supports_lora, 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",
|
|
"SupportsPP",
|
|
"supports_pp",
|
|
"SupportsTranscription",
|
|
"supports_transcription",
|
|
"SupportsV0Only",
|
|
"supports_v0_only",
|
|
]
|