[代码调试]安装Text2Image(stable diffusion)模型环境的踩坑记录
文章目录
- 一、xFormers版本问题
- 1、先确认下自己torch版本所对应的cuda版本
- 2.安装对应版本
- (1) 到对应官网下载包本地安装
- (2)代码安装
- 二、attn_mask尺寸错误
- 三、diffusers
- 四、Huggingface
配置基于stable diffusion预训练模型的环境，记录踩坑记录
一、xFormers版本问题
xformers是用来加速推理的，但是和pytorch的版本有兼容问题，需要安装对应的版本，否则会报错。
1、先确认下自己torch版本所对应的cuda版本
可以使用以下代码进行查看
import torch
print(torch.version.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一致