Open Skquark opened 7 months ago
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.
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:
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.
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.
I'm trying to run this using Xformers 0.0.25 because I have to run latest Torch 2.2.1 which Google Colab just updated to, and xformers 0.0.24 only works with Torch 2.2.0, so installing 0.0.24 takes ~5 minutes and a Restart to downgrade to torch 2.2.0. I got it working in my app at DiffusionDeluxe.com using the recommended 0.0.24 (although keeps running out of RAM), but with 0.0.25 I'm getting this error:
I'm hoping you can figure out a solution to fix the breaking change to make it compatible with both, but always nice to be using the newest versions. Thanks, tried to trace the problem down myself, but didn't understand it enough.