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

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

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

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

复制代码

进入到你下载的路径

1. conda env create -f environment.yaml
2. 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 目次下面
运行：

1. 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

来源：https://blog.csdn.net/jbzj/article/details/129408801
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！