🤗 HuggingFace Diffusers Flax TPU and PyTorch GPU for Colab - View it on GitHub