Hugging Face Blog·· 2022-10-13精选AI 评分62
如何在 JAX / Flax 上运行 Stable Diffusion
🧨 Stable Diffusion in JAX / Flax !
AI 导读
Hugging Face Diffusers 自 0.5.1 版起支持 Flax，可在 Colab、Kaggle 或 Google Cloud 的 TPU 上快速推理 Stable Diffusion。
推荐理由
Hugging Face 官方给出在 TPU 上用 JAX/Flax 跑 Stable Diffusion 的完整步骤与并行原理，可迁移到多设备推理。
来源：Hugging Face Blog · huggingface.co