HuggingFace Diffusers v0.5.0の新機能
「Diffusers v0.5.0」の新機能についてまとめました。
・Diffusers v0.5.0
前回
1. Diffusers v0.5.0 のリリースノート
情報元となる「Diffusers 0.5.0」のリリースノートは、以下で参照できます。
2. JAX/Flax Integration による超高速推論が可能に
「Stable Diffusion」のJAX/Flaxサポートが追加されました。これによって、Colab TPU (および GPU) での Stable Diffusion の超高速推論が可能になりました。
試したところ、12秒で8枚の画像を生成できました。ただし、初回の画像生成にはコンパイルなどで時間がかかります。
3. DeepSpeed による学習時のVRAMと速度の向上
HuggingFace accelerateのDeepSpeed Integrationによって、学習時のVRAMと速度が向上しました。
「DreamBooth」もGPUメモリ 8GBで学習できるようになりました。
Text2Imageのファインチューニングの例も、「DeepSpeed」と完全に互換性があります。
次回
この記事が気に入ったらサポートをしてみませんか？