4 ms·

Run Super Fast Stable Diffusion with Jax and TPUs

1 points by osanseviero 4y ago

osanseviero 4y ago
Hugging Face diffusers now supports running Stable Diffusion with JAX, allowing to generate 8 images in 8 seconds on Colab using TPUs. Tweet: https://twitter.com/psuraj28/status/1580640841583902720 https://twitter.com/psuraj28/status/1580640841583902720 Blog announcement: https://huggingface.co/blog/stable_diffusion_jax https://huggingface.co/blog/stable_diffusion_jax