search

p1atdev / LECO

Low-rank adaptation for Erasing COncepts from diffusion models.
https://arxiv.org/abs/2303.07345
Apache License 2.0
307 stars 23 forks source link

Issue with training #38

Open davizca opened 7 months ago

davizca commented 7 months ago

Hi!

1fbd895b06b2f8f720be7175a48c10be I'm having this problem trainng LECO... any solution to this?

Thanks in advance

ChrisWang13 commented 5 months ago

Obviously something is wrong with the version of xformers.

You need to verify the version of your cuda and Pytorch to check whether they are compatible with your current xformers.

In setup, you should try: pip install torch torchvision --index-url https://download.pytorch.org/whl/YOUR_CUDA_VERSION