--- license: openrail library_name: diffusers tags: - TPU - JAX - Flax - stable-diffusion - text-to-image language: - en datasets: - camenduru/plushies inference: false --- ๐Ÿงจ pytorch checkpoints, maybe this is the first ever model trained with TPUs and converted to PyTorch hehe ๐ŸŽŠ๐ŸŽ‰
Trained with google cloud TPUs. ``` Runtime: 3h 26m 44s Steps: 18000 Precision: bf16 Learning Rate: 1e-6 ``` ![plushies](/camenduru/plushies/resolve/main/samples.jpg)