stable diffusion をCPUマシンで動かす

環境情報

  • Ubuntu 22.04

  • CPUのみ、GPUなし

  • 筐体は2016年だかのmacbook pro

結論

うごかなかった・・・

本家の場合

上記スペックのMacでは動作しない模様。
CPU非対応なのか、RAMが足りないのかなどの調査まではやっていない。
※モデルが4GB程度もあるので、RAMメモリ不足が濃厚と思うが・・・

(ldm) @MacBookPro:~/src/stable-diffusion$ python scripts/txt2img.py --prompt "a photograph of an astronaut riding a horse" --plms 
Global seed set to 42
Loading model from models/ldm/stable-diffusion-v1/model.ckpt
Global Step: 470000
LatentDiffusion: Running in eps-prediction mode
強制終了

有志のMac OS用ソースの場合

こちらのサイトでMAC(CPU)で動かす方法を見つける。
※MAC対応にforkされたリポジトリのソースを使う。
上記と変わらずエラーで強制終了する。

導入の大枠

依存関係が以下のようになっているので、逆順に進めていく形

  1. Stable Diffusionの導入

  2. Condaが必要

  3. Anacondaが必要

  4. pythonが必要

pythonの導入

  1. 導入済みか確認(入っていた)

@MacBookPro:~$ python3
Python 3.10.4 (main, Jun 29 2022, 12:14:53) [GCC 11.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> 

Anacondaの導入

公式サイトを参考にする。

Stable Diffusionの導入

公式サイトを参考にする。

重みファイルの入手

huggingfaceにサインアップし、Access to the repositoryを押して、連絡先を共有することに同意してからDLする。

この記事が気に入ったらサポートをしてみませんか?