跳到正文
Hugging Face 社区博客·· 2022-10-13AI 评分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 社区博客 · huggingface.co