[代码调试]安装Text2Image(stable diffusion)模型环境的踩坑记录

配置基于stable diffusion预训练模型的环境，记录踩坑记录

一、xFormers版本问题

xformers是用来加速推理的，但是和pytorch的版本有兼容问题，需要安装对应的版本，否则会报错。

1、先确认下自己torch版本所对应的cuda版本

可以使用以下代码进行查看

import torch
print(torch.version.cuda)

然后命令行 pip list 查看torch 版本，根据 torch 和cuda版本进本xformer版本的查找

1.1 查找方式1，官网查找

然后确认xformer版本，进去网站https://github.com/facebookresearch/xformers/tree/v0.0.27
点击不同的tag

下面有对应的torch版本

1.1 查找方式2，conda库查找（个人使用，比较准）

 https://anaconda.org/xformers/xformers/files

上面每个包有对应的版本信息，xformer python pytorch cuda

2.安装对应版本

安装方式有三种

(1) 到对应官网下载包本地安装

如果是pip安装到这里

 https://download.pytorch.org/whl/cu118

如果是conda安装到这里

 https://anaconda.org/xformers/xformers/files

上面有不同版本及对应关系，如果不知道当前版本需要安装哪个xFormers版本，可以在这找找

(2)代码安装

上面网站conda 查找到需要安装的版本，直接pip安装

pip3 install xformers==0.0.26.post1 

或者直接代码安装

pip3 install -U xformers --index-url https://download.pytorch.org/whl/cu118

后面的cu118换成自己的cuda版本，但是有可能torch版本会给你换掉，注意看下

二、attn_mask尺寸错误

出现这个问题RuntimeError: The shape of the 2D attn_mask is torch.Size([77, 77]), but should be (1, 1)

一般需要查看transformers 和 open-clip-torch 版本，让其和官方requirements.txt一致。反正就是升降级

三、diffusers

如果发现找不到diffusers，可以尝试安装一下更低or更高的版本，最好和官方requirements.txt一致

四、Huggingface

如果发现找不到Huggingface，可以尝试安装一下更低or更高的版本，最好和官方requirements.txt一致