Diffusion Model DDPM: Denoising Diffusion Probabilistic Models
加速Clip 、 UNet 、 ControlNet 、VAE Encoder 和 Decoder
初赛是利用 TensorRT 加速带有 ControlNet 的 Stable Diffusion canny2image pipeline,以优化后的运行时间和出图质量作为主要排名依据。
流程
img = resize_image(HWC3(input_image), image_resolution)
# H, W, C = img.shape uint8
参考
- NVIDIA TensorRT Hackathon 2023 —— 生成式AI模型优化赛_算法大赛_赛题与数据_天池大赛-阿里云天池
- https://maplesyrup.fanbox.cc/posts/5378959
- GitHub - kohya-ss/sd-webui-additional-networks
- Make stable diffusion up to 100% faster with Memory Efficient Attention | PhotoRoom
- Making stable diffusion 25% faster using TensorRT | PhotoRoom
- runwayml/stable-diffusion-v1-5 · Hugging Face
- CLIP fp16溢出解决办法_天池技术圈-阿里云天池