Stable Diffusionをローカルで動かしてみた

本家を試してみたが、GPUのメモリ不足エラーが出たので省メモリ版で試した。

要インストール

anaconda

https://www.anaconda.com/

cuda

https://developer.nvidia.com/cuda-toolkit

V10.0で動いた

導入

stable-diffusion インストール

1clone https://github.com/basujindal/stable-diffusion.git
2cd stable-diffusion
3conda env create -f environment.yaml
4conda activate ldm

model ダウンロード

こちらでユーザ登録後、規約に同意

任意のディレクトリでclone

1git clone https://huggingface.co/CompVis/stable-diffusion-v-1-4-original

model配置

stable-diffusion-v-1-4-original -> stable-diffusion-v-1 にリネーム

sd-v1-4.ckpt -> model.ckpt にリネーム

stable-diffusion-v-1 -> stable-diffusion/models/ldm にコピー

実行

1python optimizedSD/optimized_txt2img.py --prompt "Cyberpunk style image of a Telsa car reflection in rain" --H 512 --W 512 --seed 27 --n_iter 2 --n_samples 10 --ddim_steps 50