xFormers可以加快图像生成速度(几乎快两倍)并使用更少的 GPU 内存。它被广泛使用并且工作得很好,但与您之前生成的图像相比,它有时会生成不同的图像(对于相同的提示+设置)。
重要的!!xFormers 仅适用于配备 NVIDIA GPU 的 PC。它在纯 CPU 计算机或 M1/M2 Mac 上没有用。
安装 xFormers
- 双击
Developer Console.cmd
(在 Windows 上)或运行./developer_console.sh
(在 Linux/Mac 上) - 运行
python -m pip show torch torchvision
(重要:请记下这两个版本,以防您需要恢复到它们) - 运行以下命令:
python -m pip install --upgrade torch==2.0.1+cu118 torchvision==0.15.2+cu118 xformers==0.0.22 --extra-index-url https://download.pytorch.org/whl/cu118
或使用pytorch 中的命令生成器(位于页面顶部)。这应该足够了,您现在可以像平常一样启动 Easy Diffusion。
删除/卸载 xFormers
如果您遇到任何问题,请通过以下方式回滚:
- 双击
Developer Console.cmd
(在 Windows 上)或运行./developer_console.sh
(在 Linux/Mac 上) - 跑步
python -m pip uninstall xformers
- (可选)将
torch
和torchvision
版本恢复为之前的版本:
- Windows 或 Linux:运行
python -m pip install torch==version_noted_above torchvision==version_noted_above --index-url https://download.pytorch.org/whl/cu116
- 苹果机:运行
python -m pip install torch==version_noted_above torchvision==version_noted_above