Hugging Face Blog·· 2022-10-13AI 评分55
Hugging Face Diffusers 教程：用 JAX/Flax 在 TPU 上运行 Stable Diffusion
🧨 Stable Diffusion in JAX / Flax !
AI 导读
Diffusers 自 0.5.1 起支持 Flax，官方教程演示如何在 Google TPU 上用 FlaxStableDiffusionPipeline 运行 CompVis/stable-diffusion-v1-4 推理。
来源：Hugging Face Blog · huggingface.co