Hugging Face·· 2022-10-13精选AI 评分66
Hugging Face 发布 JAX/Flax 版 Stable Diffusion 推理实践指南
🧨 Stable Diffusion in JAX / Flax !
AI 导读
Hugging Face Diffusers 在 0.5.1 版本加入对 JAX 和 Flax 的支持，允许开发者在 Google TPU 设备上高效并行运行 Stable Diffusion 推理。
推荐理由
原文给出了利用 JAX 的 pmap 与 TPU 多核并行加速图像生成的具体范式，读者可直接复用其编译与分发逻辑。
来源：Hugging Face · huggingface.co