Hugging Face Blog·· 2023-10-03AI 评分42
Hugging Face Diffusers 支持在 Cloud TPU v5e 上用 JAX 加速 Stable Diffusion XL 推理
🧨 Accelerating Stable Diffusion XL Inference with JAX on Cloud TPU v5e
AI 导读
Hugging Face Diffusers 现已支持在 Cloud TPU v5e 上通过 JAX 运行 Stable Diffusion XL 推理。该方案借助 JAX 的 jit 编译与 XLA 并行能力，在多个 TPU v5e-4 实例上约 4 秒生成四张 1024×1024 图像，其中实际生成时间约 2.3s。
来源：Hugging Face Blog · huggingface.co