mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-14 06:27:45 +08:00
c6e2111f4e
* feat: enhance trtllm serve multimodal 1. made the load_image and load_video asynchronous 2. add image_encoded input support to be compatible with genai-perf 3. support text-only on multimodal mdoels(currently, Qwen2-VL & Qwen2.5-VL) Signed-off-by: yechank <161688079+yechank-nvidia@users.noreply.github.com> * add test Signed-off-by: yechank <161688079+yechank-nvidia@users.noreply.github.com> * fix bandit Signed-off-by: yechank <161688079+yechank-nvidia@users.noreply.github.com> * trimming uils Signed-off-by: yechank <161688079+yechank-nvidia@users.noreply.github.com> * trimming for test Signed-off-by: yechank <161688079+yechank-nvidia@users.noreply.github.com> * genai perf command fix Signed-off-by: yechank <161688079+yechank-nvidia@users.noreply.github.com> * command fix Signed-off-by: yechank <161688079+yechank-nvidia@users.noreply.github.com> * refactor chat_utils Signed-off-by: yechank <161688079+yechank-nvidia@users.noreply.github.com> * stress test genai-perf command Signed-off-by: yechank <161688079+yechank-nvidia@users.noreply.github.com> --------- Signed-off-by: yechank <161688079+yechank-nvidia@users.noreply.github.com>
18 lines
967 B
Python
18 lines
967 B
Python
from .data import PromptInputs, TextPrompt, TokensPrompt, prompt_inputs
|
|
from .registry import (ExtraProcessedInputs, InputProcessor,
|
|
create_input_processor, register_input_processor)
|
|
from .utils import (INPUT_FORMATTER_MAP, async_load_image, async_load_video,
|
|
default_image_loader, default_video_loader,
|
|
encode_base64_content_from_url, format_generic_input,
|
|
format_qwen2_vl_input, format_vila_input, load_image,
|
|
load_video)
|
|
|
|
__all__ = [
|
|
"PromptInputs", "prompt_inputs", "TextPrompt", "TokensPrompt",
|
|
"InputProcessor", "create_input_processor", "register_input_processor",
|
|
"ExtraProcessedInputs", "load_image", "load_video", "async_load_image",
|
|
"async_load_video", "INPUT_FORMATTER_MAP", "default_image_loader",
|
|
"default_video_loader", "format_vila_input", "format_generic_input",
|
|
"format_qwen2_vl_input", "encode_base64_content_from_url"
|
|
]
|