comfyUI独立安装包中,安装xformers,加速图像生成
Xformers是什么
xFormers 是一个由 Meta(原 Facebook Research)开发的基于 PyTorch 的优化库,专注于提升 Transformer 模型的计算效率和显存利用率。它通过定制化的 CUDA 内核、模块化组件和前沿算法优化,为深度学习任务(尤其是生成式模型如 Stable Diffusion)提供加速支持。
xFormers 提供了多种优化的注意力机制(如 Memory-Efficient Attention 和 Flash Attention),通过减少显存占用和加速矩阵运算,显著提升模型推理和训练效率。例如,在生成图像时,显存消耗可降低 30%-50%,速度提升可达 2 倍。
xFormers 是 ComfyUI 用户提升生成效率和稳定性的关键工具,尤其适合需要处理复杂工作流或显存有限的场景。
如果没有安装,可能会报类似这样的错误:
no module 'xformers'.Processing without...
这个报错出现,并不影响comfyui可以正常运行,图像还是可以正常生成的,只是速度会慢一些,如果你的显卡本身显存不大,还是建议更新安装下的。
如何安装
注意:xformers版本跟你电脑的pytorch和cuda版本关系非常大,如果不兼容有可能直接把你当前的运行环境搞崩溃,所以如果崩溃了,而自己又不懂如何修复,可能要重新安装comfyui。
首先需要到xformers官方的git上查看下版本兼容的问题:
GitHub - facebookresearch/xformers: Hackable and optimized Transformers building blocks, supporting a composable construction.
https://github.com/facebookresearch/xformers
比如说,这边是12.4的cuda,所以要找到这行命令:
cuda 12.4 version
但是,请勿直接安装:
如果是这样安装,系统会自动更新torch到最新的2.6,然后启动comfyui你就会遇到这样的提示:
comfyui环境就崩溃了,而且会遇到跟torchvison等几个package冲突的问题,很麻烦。
多次测试后,找到一个比较靠谱的组合:
torch2.5.1+cu124+xformers 0.0.28.post3版本。
命令行输入:python -m pip install -U xformers==0.0.28.post3 --index-url https://download.pytorch.org/whl/cu124 然后回车