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 506 forks source link

Dreambooth in flax? #215

Open droidXrobot opened 1 year ago

droidXrobot commented 1 year ago

Describe the bug

I’ve tried just about everything to get dreambooth running with flax but get errors each time. Has anyone got it working with flax?

Reproduction

I’ve tried an A100 and TPU in Colab and followed the directions in the readme

Logs

No response

System Info

Colab, using requirements in your dreambooth readme