search

NVIDIA / TensorRT

NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source components of TensorRT.
https://developer.nvidia.com/tensorrt
Apache License 2.0
10.81k stars 2.13k forks source link

How to support referenceNet+unet? #4226

Open songh11 opened 2 weeks ago

songh11 commented 2 weeks ago

I want use tensorrt to accelerate https://github.com/MooreThreads/Moore-AnimateAnyone, but I don't know how to support referenceNet.Can you give me some advice?

yuanyao-nv commented 2 weeks ago

You can try exporting it as an ONNX model and run with tensorrt: https://docs.nvidia.com/deeplearning/tensorrt/developer-guide/index.html#onnx-intro