Error with Xformers newest version

[thu-ml]

Error with Xformers newest version

[Riyue120]

I have tested a new environment (torch2.2.2+cu118), and used the newest xformers.
You can use the offcial method to get the newest version.

# cuda 11.8 version
pip3 install -U xformers --index-url https://download.pytorch.org/whl/cu118

Here are my steps:

# python.version //3.8
# print(torch.__version__) //2.2.2+cu118
git clone https://github.com/thu-ml/CRM.git
cd CRM
pip install -r requirements.txt
pip install ninja
pip3 install -U xformers --index-url https://download.pytorch.org/whl/cu118
pip install git+https://github.com/NVlabs/nvdiffrast
# other package(forget)

I made a docker image in CodeWithGPU platform, Here are the page: Page Link.
It is under reviewing and will take two days.

docker pull registry.cn-beijing.aliyuncs.com/codewithgpu2/thu-ml-crm:x00G39kpFb

Along the way, I also encountered some strange things:

1. I found the .pth and .bin files will be storaged in the ./.cache/huggingface/hub/model*, it is difficult for me to move them directly.

2. I can not run the code python run.py "imgfile" in juypter, it will raise errors about Connection with huggingface.co, but I can run this code in the terminal.

[thuwzy]

My testing environment is torch==1.13.0+cu117. Maybe you can try torch 1.x? I am not sure whether will torch 2.x work.