一. 什么是 xFormers

🚅webui 开启 xformers 提速 100% - 图1

Xformers library is an optional way to speedup your image generation. This optimization is only available for nvidia gpus, it speeds up image generation and lowers vram usage at the cost of producing non-deterministic results.

Xformers 是 Facebook 出品的一款加速 Transformers 的工具库,xformers 可以加速图片生成,并显著减少显存占用,是小显存玩家的福音,代价是可能生成不确定性的结果,注意该选项仅可用于 Nvidia 显卡。详见 webui 的简介:

Xformers

二. 安装步骤

进入虚拟环境

为了避免和其他环境库的版本产生冲突,webui 有一套自己的虚拟环境,首先我们需要激活 webui 的虚拟环境,然后在这套虚拟环境中更新。stable-diffusion-webui 文件【Shift 右键】:

🚅webui 开启 xformers 提速 100% - 图2

点击”在此处打开 Powershell 窗口(S)”,进入 PowerShell 终端,执行命令:

  1. $ .\venv\Scripts\Activate.ps1

如图,进入虚拟环境:

🚅webui 开启 xformers 提速 100% - 图3

安装或更新 xformers 版本

这里本机环境是 torch: 2.0.0+cu118,xformers 最新版本已经支持 2.0 torch,所以这里直接安装最新的 0.0.19 版本,可以再这里查看 xformers 最新版本:

xformers

  1. pip install xformers==0.0.19

🚅webui 开启 xformers 提速 100% - 图4

在 webui 中开启 xformers

  1. $ vim webui-user.bat

🚅webui 开启 xformers 提速 100% - 图5

三. 使用效果

开启前:耗时 17.36,显存 11.5 GB

🚅webui 开启 xformers 提速 100% - 图6

开启后:耗时 8.45s,显存 9.6 GB

🚅webui 开启 xformers 提速 100% - 图7