Spaces:
Runtime error
Runtime error
# 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() | |