download translated onnx model
vim_h trt model tested on GPU 3060 Ti, tests ok.
tutorials.ipynb
tutorials_vim_h.ipynb
how_to_export_vim_h_model.ipynb
# Divide the embeding model into 2 parts, named as part1 and part2
# The decoder model remains unchanged
# export_embedding_model_part_1()
# export_embedding_model_part_2()
# export_sam_h_model()
# onnx_model_example2()
you can not set FP16 value in export.h, if gpu have enough mems.