diffusers加速文生图速度；stable-diffusion、PixArt-α

2024-01-07 17:13:31

参考：
https://pytorch.org/blog/accelerating-generative-ai-3/
https://colab.research.google.com/drive/1jZ5UZXk7tcpTfVwnX33dDuefNMcnW9ME?usp=sharing#scrollTo=jueYhY5YMe22

大概GPU资源8G-16G；另外模型资源下载慢可以在国内镜像：https://aifasthub.com/
在这里插入图片描述

1、加速代码

能加速到2秒左右

from diffusers import StableDiffusionXLPipeline
import torch

pipe = StableDiffusionXLPipeline.from_pretrained(
    "stabilityai/stable-diffusion-xl-base-1.0<

文章来源:https://blog.csdn.net/weixin_42357472/article/details/135395214
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。