Stable Diffusion 在 JAX / Flax 上的推理教程

Hugging Face Diffusers 自 0.5.1 版本起支持 Flax，可在 Google TPU 上运行 Stable Diffusion 推理，教程演示了在 Colab 等环境用 8 个 TPU 加速器并行生成 8 张图片。模型权重以 bf16 版本提供，采用 CreativeML OpenRAIL-M 开放许可，允许再分发与商用，但需向用户附带相同使用限制。