この記事ではDiffuserのインストール方法を記載します．
目次
インストール方法
PyTorch用の環境を構築
まずは，Pytorchを動作させるための環境を構築します
PyTorchのインストール関連記事にまとめてあります．
仮想環境の構築
仮想環境を用意します．この手順は不要であればスキップ可能です．
mkdir venv_diffuser
python -m venv venv_diffuser
source venv_diffuser/bin/activate
Diffuserのインストール
PyTorchも依存関係として自動的にインストールされます．
pip install --upgrade diffusers[torch] accelerate transformers
accelerateとtransformersも同時にインストールします．
動作確認
動作確認はStable Diffusionモデルを使用して画像を生成します．
下記のようなファイルを作成し，実行します．
main.py
from diffusers import DiffusionPipeline
pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
pipeline.to("cuda")
image = pipeline("An image of a squirrel in Picasso style").images[0]
image.save("image_of_squirrel_painting.png")
実行すると下記のような画像が生成されます．
python main.py
image_of_squirrel_painting.png
