TensorRT-LLMs/tensorrt_llm/inputs/__init__.py
2025-02-25 21:21:49 +08:00

11 lines
511 B
Python

from .data import PromptInputs, TextPrompt, TokensPrompt, prompt_inputs
from .registry import (ExtraProcessedInputs, InputProcessor,
create_input_processor, register_input_processor)
from .utils import get_hf_image_processor, load_image, load_video
__all__ = [
"PromptInputs", "prompt_inputs", "TextPrompt", "TokensPrompt",
"InputProcessor", "create_input_processor", "register_input_processor",
"ExtraProcessedInputs", "get_hf_image_processor", "load_image", "load_video"
]