安装 xFormers
# cuda 11.8 version pip install -U xformers --index-url https://download.pytorch.org/whl/cu118 # cuda 12.1 version pip install -U xformers --index-url https://download.pytorch.org/whl/cu121
# cuda 11.8 version pip install -U xformers --index-url https://download.pytorch.org/whl/cu118 # cuda 12.1 version pip install -U xformers --index-url https://download.pytorch.org/whl/cu121