Convert and Inference T2I-Adapter models in ONNX and TensorRT. Polygraphy is used to convert from onnx to tensorrt and create engine file.
Original Image | Converted by Canny Adapter |
---|---|
convert_stable_diffusion_checkpoint_to_onnx.py
@misc{https://doi.org/10.48550/arxiv.2302.08453,
doi = {10.48550/ARXIV.2302.08453},
url = {https://arxiv.org/abs/2302.08453},
author = {Mou, Chong and Wang, Xintao and Xie, Liangbin and Zhang, Jian and Qi, Zhongang and Shan, Ying and Qie, Xiaohu},
keywords = {Computer Vision and Pattern Recognition (cs.CV), Artificial Intelligence (cs.AI), Machine Learning (cs.LG), Multimedia (cs.MM), FOS: Computer and information sciences, FOS: Computer and information sciences},
title = {T2I-Adapter: Learning Adapters to Dig out More Controllable Ability for Text-to-Image Diffusion Models},
publisher = {arXiv},
year = {2023},
copyright = {Creative Commons Attribution 4.0 International}
}