mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-14 06:27:45 +08:00
13 lines
379 B
Python
13 lines
379 B
Python
import argparse
|
|
|
|
from tensorrt_llm.tools.multimodal_builder import (MultimodalEngineBuilder,
|
|
add_multimodal_arguments)
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser()
|
|
parser = add_multimodal_arguments(parser)
|
|
args = parser.parse_args()
|
|
|
|
builder = MultimodalEngineBuilder(args)
|
|
builder.build()
|