raksa-the-wildcats
first commit
383af88
raw
history blame contribute delete
482 Bytes
# 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()