15 lines
482 B
Python
15 lines
482 B
Python
# copied from: https://github.com/NVIDIA/TensorRT-LLM/blob/v0.18.2/examples/multimodal/build_visual_engine.py
|
|
|
|
import argparse
|
|
|
|
from tensorrt_llm.tools.multimodal_builder import (VisionEngineBuilder,
|
|
add_multimodal_arguments)
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser()
|
|
parser = add_multimodal_arguments(parser)
|
|
args = parser.parse_args()
|
|
|
|
builder = VisionEngineBuilder(args)
|
|
builder.build()
|