Huggingface Stable Diffusionによる画像生成

このコードは、「diffusers」ライブラリの「StableDiffusionPipeline」を使用して、与えられたプロンプトに基づいて画像を生成します。

この場合、プロンプトは「少女が月の表面を歩いています。」です。「StableDiffusionPipeline」は、拡散とディープラーニングを組み合わせた生成モデルの一種で、高品質の画像を生成することができます。

!pip install --upgrade diffusers transformers scipy

from diffusers import StableDiffusionPipeline
import torch

model_id = "runwayml/stable-diffusion-v1-5"
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
pipe = pipe.to("cuda")

prompt = "A girl is walking on the surface of the moon."

for i in range(1):
    image = pipe(prompt).images[0]  
    image.save("moongirl_" + str(i) + ".png")

最初のステップは、「!pip install --upgrade diffusers transformers scipy」を使用して、必要なライブラリ「diffusers」、「transformers」、「scipy」をインストールすることです。

次に、コードは「runwayml/stable-diffusion-v1-5」から「StableDiffusionPipeline」をロードし、トーチのデータ型を「torch.float16」に設定します。パイプラインは、より速い処理のためにGPUにします。

最後に、コードはプロンプトを使用して1枚の画像を生成し、「moongirl_0.png」というファイルに保存します。複数枚作成したいときは、1という数字を作成したい枚数に変更してください。