🧨 Stable Diffusion in JAX / Flax !

Hugging Face 推出 JAX/Flax 版 Stable Diffusion，支援 TPU 與 GPU 超高速並行圖像生成。

Hugging Face 宣布在 diffusers 庫中支援 JAX 和 Flax 版本的 Stable Diffusion。藉由 JAX 的 XLA 編譯與硬體加速，開發者可以在 TPU 和 GPU 上實現極致的推理速度，並輕鬆透過並行運算（pmap）在數秒內生成大量高品質圖片。這為大規模圖像生成與研究提供了全新的高效能選擇。

摘要由 AI 整理,以原文為準。