Hugging Face Blog·· 2022-10-13精选AI 评分62
Hugging Face 教程：用 JAX / Flax 在 TPU 上运行 Stable Diffusion
🧨 Stable Diffusion in JAX / Flax !
AI 导读
Hugging Face Diffusers 自 0.5.1 版本起支持 Flax，可在 Colab、Kaggle 或 Google Cloud Platform 的 TPU 上实现快速推理。
推荐理由
Hugging Face 官方给出在 TPU 上用 JAX/Flax 跑 Stable Diffusion 的完整流程，含并行与编译耗时数据。
来源：Hugging Face Blog · huggingface.co