巷で話題の『stable diffusion』、せっかくM1 MBP買ったのでローカルで環境構築試してみました。
Twitterやブログで勝手に知恵をお借りしましてとりあえず画像生成できるところまで辿り着きました〜・・・
PCスペック
- MacBook Pro（14インチ、2021）
- macOS Monterey
- Apple M1 Max
- メモリ32GB
参考にさせていただいた記事、ツイート等
- stable diffusion githubリポ
- hellohazimeさんの記事（https://zenn.dev/hellohazime/articles/543fcaf862008b）
- @masakuni_itoさんのツイート
特にRuntimeError: expected scalar type BFloat16 but found Float←この辺りのエラーでずっと引っかかっていたので大変助かりましたありがとうございます。
画像生成
参考にさせていただいた記事をもとにサンプルで載っている自転車に乗る宇宙飛行士の出力例です。
python scripts/txt2img.py --prompt "a photograph of an astronaut riding a bike" --plms --precision full --n_samples 1
ポイントはオプションに--precision fullを付けること。これで先述したfloat周りのエラーを回避できるとか。
上記のコマンドでできた画像がこちら↓（画像は圧縮処理したものです）
promptメモ
今後更新していく予定。