Deng_Yao
Deng_Yao
/data/dataloader_custom.py is used to load data when you use test.py， if you want to train your own data, you may need to write a new dataloader
Thank you for your reply! I have tried Xformers==0.0.13, but one function has a different prototype than version 0.0.16, and it doesn't work. The error details are as follows: ...
I tried to run CUDA11.3 directly with the xformers version you gave in requirements.txt and it worked, please forgive me for my carelessness.
I am sorry for asking such a stupid question. After installed torch, you can fix this problem.