Apple Siliconで高速化したStable Diffusionモデル

Redditorのuqety8さんが、Stable Diffusionのパイプラインのうち、特に時間のかかるデノイジングステップをAppleのMLXフレームワークに移植し、Apple Silicon搭載のMacBook Pro M3 Proで高速化に成功したことを報告しています。MLXバージョンでは1024x1024の画像生成が1ステップあたり19秒と、PyTorch MPSの720x720の20秒と比べても高速化が確認できたとのことです。今後はさらなる高速化を目指して、他のコンポーネントもMLXに移植する予定とのことです。