模型训练库缺失 ImportError: No xformers

发现问题

借助Lora训练Diffusion模型的时候，出现了报错：

ImportError: No xformers

分析问题

安装对应版本的Transformer模型即可解决问题，参考：https://github.com/Akegarasu/lora-scripts/issues/422#issuecomment-2127945481

解决问题

找到当前运行训练任务的机器，确定对应的版本：

import torch
pirnt(torch.__version__)

>>> 2.3.1 + cu121

从输出定位 torch版本 和 cuda版本，从 https://github.com/facebookresearch/xformers?tab=readme-ov-file#installing-xformers 找 torch + cuda + xformers 三者关系都能对应上的版本。

# cuda 12.1 version
pip3 install -U xformers --index-url https://download.pytorch.org/whl/cu121

安装成功，问题解决。