File size: 482 Bytes
383af88
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
# 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()