Stable diffusion环境部署搭建

SEO教程

正在检查是否收录...

一言准备中...

1、安装Nvidia驱动、cuda版本等

略

2、安装anaconda环境

略

3、安装git

#git工具的安装参考下面命令 apt-get update -y a pt-get upgrade -y apt install git

4、下载源码和模型参数文件

#1、下载源码 git clone https://github.com/CompVis/stable-diffusion.git #2、下载模型参数文件，以下示例为v1.4版本，大约7GB wget https://xujianhua-bj.tos-cn-beijing.volces.com/sd-v1-4-full-ema.ckpt #3、设置模型参数存放位置 #git下来的stable-diffusion文件夹路径中创建stable-diffusion-v1目录 mkdir -p /root/stable-diffusion/models/ldm/stable-diffusion-v1/ #将下载的模型参数文件改名为 model.ckpt 并放在 stable-diffusion-v1 目录下 ln -s /root/sd-v1-4-full-ema.ckpt /root/stable-diffusion/models/ldm/stable-diffusion-v1/model.ckpt

5、Anaconda创建 ldm 虚拟环境

#切换清华源，便于后续生成环境，下载python包 pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

//注意：火山引擎FT的只开github的加速，不开http/https代理。因为设置http/https代理，执行conda env create -f environment.yaml会报错。

#进入之前的源码解压目录 cd stable-diffusion conda env create -f environment.yaml conda activate ldm

#environment.yaml的内容 name: ldm channels: - pytorch - defaults dependencies: - python=3.8.5 - pip=20.3 - cudatoolkit=11.3 - pytorch=1.11.0 - torchvision=0.12.0 - numpy=1.19.2 - pip: - albumentations==0.4.3 - diffusers - opencv-python==4.1.2.30 - pudb==2019.2 - invisible-watermark - imageio==2.9.0 - imageio-ffmpeg==0.4.2 - pytorch-lightning==1.4.2 - omegaconf==2.1.1 - test-tube>=0.7.5 - streamlit>=0.73.1 - einops==0.3.0 - torch-fidelity==0.3.0 - transformers==4.19.2 - torchmetrics==0.6.0 - kornia==0.6 - -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers - -e git+https://github.com/openai/CLIP.git@main#egg=clip - -e . #pip install -e git+https://github.com/openai/CLIP.git@main#egg=clip #pip install -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers #pip install -e. #因为环境无法稳定访问github，导致相关包安装失败 #建议执行虚拟环境创建前，开启FT的github加速。

6、文生图

python scripts/txt2img.py --prompt "egg in the pocket" --plms --n_samples 2 #因为环境无法稳定访问github，导致相关包安装失败， #执行scripts/txt2img.py过程会报无数次 ModuleNotFoundError: No module named XX的错误 #解决办法：pip install 对应的模块名称，解决不了google #1、最后pip install方式因为版本不对，跑.py还是会失败，所以建议开通FT，能够访问境外网站和github，避免因为版本不对，因为造成无法使用问题 #2、即便开了github的网际快车，还是会遇到报错的问题，需要再开始FT的Http/https的代理。 #都设置后，还会出现包未安装的现象，执行如下命令进行安装。 #apt-get install libsm6 #apt-get install -y libxrender-dev #生成的文件在：/root/stable-diffusion/outputs/txt2img-samples/samples/

gitcodegithubdiffusiontransformertransformerspythoncliamlclippytorchurlopenai虚拟环境createscriptpython包clonetpu文生图文件夹ivanumpylmsidegooglepromptstreamlit