Is there a TensorRT ControlNet Demo? About Img2Img

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.87k stars 2.14k forks source link

Is there a TensorRT ControlNet Demo? About Img2Img #3780

Open designInno opened 7 months ago

designInno commented 7 months ago

Is there a TensorRT ControlNet Demo? About Img2Img

'demo/Diffusion/demo_controlnet.py' seems to be just a txt2Img Demo

lix19937 commented 7 months ago

ref https://github.com/NVIDIA/TensorRT/blob/release/10.0/demo/Diffusion/demo_img2img.py

Generate an image with ControlNet guided by image(s) and text prompt(s)

python3 demo_controlnet.py "Stormtrooper's lecture in beautiful lecture hall" --controlnet-type depth --hf-token=$HF_TOKEN --denoising-steps 20 --onnx-dir=onnx-cnet-depth --engine-dir=engine-cnet-depth