当前位置:   article > 正文

PixArt LCM加速文生图模型diffusers使用案例_pixart-alpha 支持图生图吗

pixart-alpha 支持图生图吗

参考:https://huggingface.co/PixArt-alpha/PixArt-LCM-XL-2-1024-MS

使用下来这个模型效果速度算最不错的

下载模型:(1080p图片生成,LCM模型是lcm加速)
PixArt-LCM-XL-2-1024-MS

PixArt-XL-2-1024-MS

1、下载

国内快速下载参考:https://blog.csdn.net/weixin_42357472/article/details/132663693

2、使用PixArt-LCM-XL-2-1024-MS(速度10s左右可以出图)

cpu:

1)测试效率图片效果要越好,提高num_inference_steps=24 数值
2)enable_model_cpu_offload(),实际上最后还是使用了gpu的;是在gpu资源不足可以这种方式来进行推理使用:这段代码将整个SD模型或SD模型的部分模块的权重加载到CPU中,并在推理时再将所需的权重加载到GPU,以实现更好的性能和灵活性。

import torch
from diffusers import PixArtAlphaPipeline

# only 1024-MS version is supported for now
  • 1
  • 2
  • 3
  • 4
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/IT小白/article/detail/708050
推荐阅读
  

闽ICP备14008679号