AttributeError: module ‘xformers.ops’ has no attribute ‘memory_efficient_attention’

问题场景

用户在 ComfyUI 中运行 flux.2 klein workflow bundle，执行过程中弹出报错。从 Traceback 看，问题发生在 VAE 编码路径：comfy/sd.py 调用 self.first_stage_model.encode(pixels_in)，随后进入 comfy/ldm/modules/diffusionmodules/model.py 的 xformers_attention，在调用 xformers.ops.memory_efficient_attention(q, k, v, attn_bias=None) 时抛出 AttributeError。

Issue 正文中的复现步骤显示，用户先进入 ComfyUI 目录并激活 .venv，然后进入本地 xformers 源码目录，用以下命令从 GitHub main 分支安装 xformers：

pip install -v --no-build-isolation -U git+https://github.com/facebookresearch/xformers.git@main#egg=xformers

之后回到 ComfyUI 目录运行 python main.py --listen --enable-manager，再触发 flux.2 klein workflow。

报错原文

[ERROR] !!! Exception during processing !!! module 'xformers.ops' has no attribute 'memory_efficient_attention'
[ERROR] Traceback (most recent call last):
  File "/home/edison/Downloads/ComfyUI/execution.py", line 542, in execute
    output_data, output_ui, has_subgraph, has_pending_tasks = await get_output_data(prompt_id, unique_id, obj, input_data_all, execution_block_cb=execution_block_cb, pre_execute_cb=pre_execute_cb, v3_data=v3_data)
  ...
  File "/home/edison/Downloads/ComfyUI/comfy/ldm/modules/diffusionmodules/model.py", line 298, in xformers_attention
    out = xformers.ops.memory_efficient_attention(q, k, v, attn_bias=None)
          ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
AttributeError: module 'xformers.ops' has no attribute 'memory_efficient_attention'

原因分析

从下游 Traceback 只能确定一件事：当前被导入的 xformers.ops 中不具备 memory_efficient_attention 这个属性。维护者指出，需要把问题“先缩小到 ComfyUI 之外”，以区分以下几类可能原因：

• Wrong/incomplete xFormers installation：xFormers 安装错误或不完整，导致 xformers.ops 没有正确暴露该函数。
• Package/API regression：所安装的 xFormers 版本存在包或 API 层面的回归。
• Missing optional backend：缺少某个可选后端依赖；维护者特别提到 mslk 包应当是 xformers 的依赖、会被自动安装，但“maybe something went wrong”。mslk 缺失目前只是怀疑方向，并非已验证根因。
• ComfyUI environment accidentally importing a different xformers：ComfyUI 环境可能误导入了另一个 xformers，而非你预期的那个安装。

维护者还表示，如果最终确认是安装失败，那么改进启动诊断可能比修改注意力代码更值得做。

环境排查

• 确认当前 python 是否来自 ComfyUI 的 .venv，避免系统 Python 与虚拟环境混用。
• 确认 xformers.__version__ 以及 xformers.ops.__file__，确认实际被导入的 xformers 来源。
• 确认 xformers.ops 是否真的具有 memory_efficient_attention 属性。
• 确认 mslk 包是否已安装；Issue 中只怀疑其可能缺失，未给出具体版本要求。
• Issue 未明确给出 CUDA、PyTorch、显卡型号和 xformers 具体版本，排查时需要自行记录这些信息。
• 确认是否从 GitHub main 分支源码安装过 xformers，以及安装过程是否完整、有无编译报错。

解决步骤

1. 先离开 ComfyUI，在同一个 ComfyUI 虚拟环境中做最小诊断，避免被 ComfyUI 的节点和模型加载干扰。维护者建议运行：

   python -m xformers.info

2. 再在同一个 Python 环境中执行最小检查：

   import xformers
   import xformers.ops as xops
   
   print(xformers.__version__)
   print(xops.__file__)
   print(hasattr(xops, "memory_efficient_attention"))

   如果 hasattr(...) 为 False，说明问题确实在 xformers 包本身或导入路径，而不是 ComfyUI 触发条件。

3. 检查 mslk 是否安装。维护者提到它应当是 xformers 的依赖、应该自动安装，但也可能安装过程出了问题。Issue 中没有给出具体安装命令或版本，因此这里只做确认，不要凭猜测补装版本。
4. 根据上一步结果分流处理：
   • 如果 xops.__file__ 指向的不是你预期从 GitHub main 安装的位置，可能是 ComfyUI 环境误导入了另一个 xformers；需要清理/统一环境后再确认。
   • 如果 xformers.ops 来自正确位置但仍缺少该属性，则可能是安装不完整或该版本存在 API 回归。
   • 如果确认是安装失败，维护者认为改进启动诊断比改注意力代码更有用；Issue 中没有给出一键修复命令。
5. 在确认环境干净、依赖完整后，再重新运行 ComfyUI 的 flux.2 klein workflow，观察是否仍在同一行抛出相同 AttributeError。

验证方法

在 ComfyUI 虚拟环境中执行上述最小诊断后，确认 print(hasattr(xops, "memory_efficient_attention")) 输出为 True；同时确认 xops.__file__ 指向你实际安装的 xformers。之后再运行 flux.2 klein workflow，若不再在 xformers_attention 处抛出 AttributeError: module 'xformers.ops' has no attribute 'memory_efficient_attention'，即可认为该报错已解决。Issue 中没有给出其他官方验证标准。

参考来源

facebookresearch/xformers #1399

AI 工具推荐

想把多个 AI 模型放在一个入口？

GamsGo AI 集成 ChatGPT、DeepSeek、Gemini、Claude、Midjourney、Veo 等常用模型，适合写作、绘图、视频和日常 AI 工作流。

了解 GamsGo AI

推广链接：通过此链接购买，我可能获得佣金，不影响你的价格。