Hugging Face 社区博客·· 2022-10-13精选AI 评分60
Hugging Face Diffusers 教程：用 JAX/Flax 在 TPU 上运行 Stable Diffusion
🧨 Stable Diffusion in JAX / Flax !
AI 导读
Hugging Face Diffusers 自 0.5.1 版本起支持 Flax，官方博客教程演示如何用 JAX/Flax 在 Google TPU 上运行 Stable Diffusion 推理。
推荐理由
Hugging Face 官方教程，演示在 TPU 上用 JAX/Flax 跑 Stable Diffusion，8 卡并行约 7 秒生成 8 张图，可直接复现。
来源：Hugging Face 社区博客 · huggingface.co