建设工程网站教程今日重大军事新闻

配置基于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一致