概要
話題のStable Diffusionすごいですね。
今まで機械学習界隈はあまり触れてこなかったのですがさすがにこれは触っておかねばと思いました。
結果がすぐに見られるのでとてもインパクトがありキラーコンテンツ感を感じています。
インストールや使い方はいろいろな方が解説してくれているので比較的すぐに実行できましたが、非エンジニアの友人には少しハードルが高かったようなのでDockerImageにしてみました。
せっかくなので利用したい方がいたら使ってもらえればと思い公開します。
ツヨツヨなグラボじゃなくても動くモデルを採用しました。
とりあえず動かすことしかやってないのでいろいろと問題があると思いますが気にならない方だけご利用ください
イメージとDockerfile
起動コマンド
docker run -it --rm -v miniconda:/opt/conda/envs -v 出力先ディレクトリ:/output -v cache:/root/.cache -v cache:/root/.huggingface --gpus all docker pull skta3569/docker-sd '画像出力用文字列' huggingfaceのトークン
「出力先ディレクトリ」にimage.pngが出力されます。
連続で動かしたら上書きしてしまうのでそのうち改善していこうと思います。