跳到正文
Hugging Face Blog·· 2023-10-03AI 评分52

Hugging Face Diffusers 支持在 Cloud TPU v5e 上用 JAX 推理 SDXL

🧨 Accelerating Stable Diffusion XL Inference with JAX on Cloud TPU v5e

AI 导读

Hugging Face Diffusers 现已支持使用 JAX 在 Cloud TPU 上部署 SDXL,实现高性能、低成本的推理。演示运行在多个 TPU v5e-4 实例上,约 4 秒生成四张 1024×1024 图像,其中实际生成时间约 2.3 秒。

来源:Hugging Face Blog · huggingface.co