Unable to load stable-diffusion-inpainting ckpt

luciddreamer-cvlab / LucidDreamer

Official code for the paper "LucidDreamer: Domain-free Generation of 3D Gaussian Splatting Scenes".

Other

1.36k stars 104 forks source link

Unable to load stable-diffusion-inpainting ckpt #41

Closed pansanity666 closed 9 months ago

pansanity666 commented 9 months ago

Hi,

I met the follow error when loading sd inpainting: OSError: Unable to load weights from pytorch checkpoint file for '../../ckpt/diff_inpaint/safety_checker/pytorch_model.bin' at '../../ckpt/diff_inpaint/safety_checker/pytorch_model.bin'. If you tried to load a PyTorch model from a TF 2.0 checkpoint, please set from_tf=True.

Do you have any idea about it?

Best,

pansanity666 commented 9 months ago

Problem solved. This is sensitive to the torch version. Using torch 2.0.1 as the readme said solves the issue.