fast-stable-diffusion: fast-dreambooth colab broken

"WARNING[XFORMERS]: xFormers can't load C++/CUDA extensions. xFormers was built for:
PyTorch 2.1.0+cu118 with CUDA 1106 (you have 2.1.0+cu121)
Python 3.9.16 (you have 3.10.12)
Please reinstall xformers (see https://github.com/facebookresearch/xformers#installing-xformers)"

I tried everything that was posted in the other issue but everything is mixed up between dreambooth and fast-a1111 and all solution posted there did not seem to fix the issue at all so I decided to start a new thread only for dreambooth. If anyone figure out the solution it would be amazing.

About this issue

  • Original URL
  • State: open
  • Created 7 months ago
  • Comments: 18 (4 by maintainers)

Most upvoted comments

!pip install lmdb
!pip install torch==2.1.2+cu121 --index-url https://download.pytorch.org/whl/cu121
!pip install -U clean-fid torchaudio torchdata torchvision torchtext
!pip install xformers --force --index-url https://download.pytorch.org/whl/cu121
!pip install -U bitsandbytes

Run this cell right before the training cell, it works for now.