stable diffusion v1及v2版本 本地部署方法、踩坑

stable diffusion v1及v2版本 本地部署方法、踩坑

    正在检查是否收录...

v1版本: 确保你至少有12G显存

git:

GitHub - CompVis/stable-diffusion: A latent text-to-image diffusion modelhttps://github.com/CompVis/stable-diffusion

git clone https://github.com/CompVis/stable-diffusion.git

进入到你下载的路径

conda env create -f environment.yaml conda activate ldm 

模型下载地址:CompVis/stable-diffusion-v-1-4-original · Hugging FaceWe’re on a journey to advance and democratize artificial intelligence through open source and open science.https://huggingface.co/CompVis/stable-diffusion-v-1-4-original

两个文件随便下载一个就可以。重命名为model.ckpt,保存到\stable-diffusion\models\ldm\stable-diffusion-v1 目录下面

运行:

python scripts/txt2img.py --prompt "a photograph of an astronaut riding a horse

提示:

Traceback (most recent call last):
  File "scripts/txt2img.py", line 22, in <module>
    from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
  File "C:\Users\x\.conda\envs\ldm\lib\site-packages\diffusers\__init__.py", line 27, in <module>
    from .pipelines import OnnxRuntimeModel
  File "C:\Users\x\.conda\envs\ldm\lib\site-packages\diffusers\pipelines\__init__.py", line 18, in <module>
    from .dance_diffusion import DanceDiffusionPipeline
  File "C:\Users\x\.conda\envs\ldm\lib\site-packages\diffusers\pipelines\dance_diffusion\__init__.py", line 1, in <module>
    from .pipeline_dance_diffusion import DanceDiffusionPipeline
  File "C:\Users\x\.conda\envs\ldm\lib\site-packages\diffusers\pipelines\dance_diffusion\pipeline_dance_diffusion.py", line 21, in <module>
    from ..pipeline_utils import AudioPipelineOutput, DiffusionPipeline
  File "C:\Users\x\.conda\envs\ldm\lib\site-packages\diffusers\pipelines\pipeline_utils.py", line 64, in <module>
    from transformers.utils import SAFE_WEIGHTS_NAME as TRANSFORMERS_SAFE_WEIGHTS_NAME
ImportError: cannot import name 'SAFE_WEIGHTS_NAME' from 'transformers.utils' (C:\Users\x\.conda\envs\ldm\lib\site-packages\transformers\utils\__init__.py)

这个确实很坑,默认里diffusers版本太高

解决方法:

conda list 查看diffusers版本,为0.14.0.

将这个删掉,

 pip uninstall diffusers

重新安装0.12.1版本

pip install diffusers==0.12.1

或者在创建环境时直接修改 environments.yaml, "diffusers" --> "diffusers==0.12.1"

如果提示:TimeoutError: [WinError 10060] 由于连接方在一段时间后没有正确答复或连接的主机没有反应,连接尝试失败。

网络问题,多试几次就好。

如果提示RuntimeError: CUDA out of memory.,加个参数 --n_samples 1

---------------------------------

V2版本地址:GitHub - Stability-AI/stablediffusion: High-Resolution Image Synthesis with Latent Diffusion Modelshttps://github.com/Stability-AI/stablediffusion

 模型下载地址:

stabilityai/stable-diffusion-2-1 · Hugging Facehttps://huggingface.co/stabilityai/stable-diffusion-2-1

diffusiongitcodegithubtransformertransformersstablediffusionamlhugging facescripthuggingfaceocrclonerapintelcreateonnxtpuwineiva重命名artificial intelligencepromptpythonurl
  • 本文作者:李琛
  • 本文链接: https://wapzz.net/post-1874.html
  • 版权声明:本博客所有文章除特别声明外,均默认采用 CC BY-NC-SA 4.0 许可协议。
本站部分内容来源于网络转载,仅供学习交流使用。如涉及版权问题,请及时联系我们,我们将第一时间处理。
文章很赞!支持一下吧 还没有人为TA充电
为TA充电
还没有人为TA充电
0
  • 支付宝打赏
    支付宝扫一扫
  • 微信打赏
    微信扫一扫
感谢支持
文章很赞!支持一下吧
关于作者
2.3W+
5
0
1
WAP站长官方

Peft库使用技巧(一):合并基座模型与Lora模型【使用Peft库微调基座模型(比如LLaMA-7B)后会得到Lora参数模块,将基座模型与Lora参数合并后才能得到完整的微调后的大模型】

上一篇

【深度学习】AUTOMATIC1111 / stable-diffusion-webui docker

下一篇
  • 复制图片
按住ctrl可打开默认菜单