search

ShivamShrirao / diffusers

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch
https://huggingface.co/docs/diffusers
Apache License 2.0
1.89k stars 509 forks source link

COLAB BOG #243

Open julianelihu opened 1 year ago

julianelihu commented 1 year ago

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