Issue with training

[davizca]

Hi!

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

Thanks in advance

[ChrisWang13]

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