Hugging Face Blog·· 2022-10-13精选AI 评分62
Hugging Face Diffusers 用 JAX / Flax 在 TPU 上运行 Stable Diffusion
🧨 Stable Diffusion in JAX / Flax !
AI 导读
Hugging Face Diffusers 自 0.5.1 版本起支持 Flax，可在 Colab、Kaggle 或 Google Cloud 的 TPU 上运行 Stable Diffusion 推理。
推荐理由
官方教程给出在 TPU 上并行跑 Stable Diffusion 的完整代码路径，可迁移到自己的推理部署。
来源：Hugging Face Blog · huggingface.co