TPU JAX Flax stable-diffusion text-to-image

flax ๐Ÿงจ checkpoints are here. ckpt and ๐Ÿงจ pytorch checkpoints here ๐ŸŽ‰๐ŸŽŠ https://huggingface.co/camenduru/plushies-pt <br/> Trained with google cloud TPUs.

Runtime: 3h 26m 44s
Steps: 18000
Precision: bf16
Learning Rate: 1e-6

plushies