COLAB BOG

[julianelihu]

Describe the bug

Hi , first Thank you for all your work, you gave us the opportunity to create

I get this error in :

https://colab.research.google.com/github/ShivamShrirao/diffusers/blob/main/examples/dreambooth/DreamBooth_Stable_Diffusion.ipynb#scrollTo=jjcSXTp-u-Eg

ValueError: Cannot load <class 'diffusers.models.autoencoder_kl.AutoencoderKL'> from digiplay/Photon_v1 because the following keys are missing:
decoder.mid_block.attentions.0.value.bias, encoder.mid_block.attentions.0.query.weight, decoder.mid_block.attentions.0.proj_attn.weight, encoder.mid_block.attentions.0.proj_attn.weight, decoder.mid_block.attentions.0.query.weight, encoder.mid_block.attentions.0.key.weight, decoder.mid_block.attentions.0.proj_attn.bias, decoder.mid_block.attentions.0.value.weight, encoder.mid_block.attentions.0.proj_attn.bias, encoder.mid_block.attentions.0.value.bias, decoder.mid_block.attentions.0.key.weight, encoder.mid_block.attentions.0.value.weight, decoder.mid_block.attentions.0.key.bias, encoder.mid_block.attentions.0.query.bias, encoder.mid_block.attentions.0.key.bias, decoder.mid_block.attentions.0.query.bias.
Please make sure to pass low_cpu_mem_usage=False and device_map=None if you want to randomly initialize those weights or else make sure your checkpoint file is correct.

Reproduction

https://colab.research.google.com/github/ShivamShrirao/diffusers/blob/main/examples/dreambooth/DreamBooth_Stable_Diffusion.ipynb#scrollTo=jjcSXTp-u-Eg

Logs

No response

System Info

Google Colab