跳到正文
原文
Hugging Face Blog·· 2022-10-13AI 评分33

Hugging Face Diffusers 0.5.1 支持 JAX / Flax 加速 Stable Diffusion 推理

🧨 Stable Diffusion in JAX / Flax !

AI 导读

Hugging Face Diffusers 0.5.1 版本新增对 JAX / Flax 的支持,允许在 Google TPU 上实现 Stable Diffusion 的极速推理。该功能利用 TPU 的 8 个并行加速器,通过参数复制和分片技术,在单次推理中同时生成 8 张图像。

来源:Hugging Face Blog · huggingface.co