HuggingFace ブログ : JAX / Flax で 🧨 Stable Diffusion ! (翻訳/解説)

【HuggingFace ブログ : JAX / Flax で 🧨 Stable Diffusion !】
🤗 Hugging Face Diffusers はバージョン 0.5.1 から Flax をサポートしています。これは Colab, Kaggle or Google Cloud Platform などで利用可能な Google TPU 上で超高速な推論を可能にします。この記事は JAX / Flax を使用して推論を実行する方法を示します。 [More]