当前位置:AIGC资讯 > AIGC > 正文

xFormers安装使用

xFormers是一个模块化和可编程的Transformer建模库,可以加速图像的生成。

这种优化仅适用于nvidia gpus,它加快了图像生成,并降低了vram的使用量,而成本产生了非确定性的结果。

下载地址:

https://github.com/facebookresearch/xformershttps://github.com/facebookresearch/xformers

启用xformers对大图及以上有极为巨大的性能提升,极大幅度降低显存占用,并且有很大可能修复黑图。代价是部分细节可能有些许改变(有可能有巨大改变)。如果不是非常吹毛求疵,强烈推荐使用xformers生成高分辨率的图片。

即使同一种子、同一参数,使用xFormers后每次出的图都有点变化,主要是物件的边缘。但是和显卡型号的改变一样,只是细节上一些变化,不对比看不出来。请自行衡量能不能接受这些改变。

linux上构建xFormers:

切换到webui根目录

source ./venv/bin/activate
cd repositories
git clone https://github.com/facebookresearch/xformers.git
cd xformers
git submodule update --init --recursive
pip install -r requirements.txt
pip install -e

Windows上构建xFormers:

安装依赖项:

1、安装VS Build Tools 2022,只需要安装Desktop development with C++

https://visualstudio.microsoft.com/downloads/?q=build+tools#build-tools-for-visual-studio-2022https://visualstudio.microsoft.com/downloads/?q=build+tools#build-tools-for-visual-studio-20222、安装Install CUDA 11.3,如果已安装Stable Diffusion WebUI,则忽略

拉取xformers源码,创建venv并激活

git clone https://github.com/facebookresearch/xformers.git
cd xformers
git submodule update --init --recursive
python -m venv venv
.\venv\scripts\activate.bat

然后安装其余的依赖项

pip install -r requirements.txt
pip install wheel

运行以下内容:

python setup.py build
python setup.py bdist_wheel

在xformers目录下,进入dist文件夹,将.whl文件复制到目录stable-diffusion-webui。

如果当前安装的是Python3.10版本,也可以直接下载打包好的whlhttps://github.com/C43H66N12O12S2/stable-diffusion-webui/releases/download/f/xformers-0.0.14.dev0-cp310-cp310-win_amd64.whlhttps://github.com/C43H66N12O12S2/stable-diffusion-webui/releases/download/f/xformers-0.0.14.dev0-cp310-cp310-win_amd64.whl

在stable-diffusion-webui目录下,安装.whl,如果名称不同,请在以下命令中更改文件名称:

.\venv\scripts\activate.bat
pip install xformers-0.0.14.dev0-cp310-cp310-win_amd64.whl

修改webui-user.bat,确保xformers被启用

set COMMANDLINE_ARGS=--xformers

最新版本的xformers不需要编译代码,可以直接通过pip安装

在stable-diffusion-webui目录下

.\venv\scripts\activate.bat
pip install xformers

更新时间 2023-11-25