Huggingface anything v3.0による画像生成

!pip install torch
!pip install --upgrade diffusers transformers scipy

from diffusers import StableDiffusionPipeline
import torch
model_id = "Linaqruf/anything-v3.0"
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
pipe = pipe.to("cuda")
 
prompt = "a girl is smiling"
image = pipe(prompt).images[0]
 
image.save("smilegirl.png")

このPythonプログラムは、「diffusers」と「transformers」パッケージを使用して画像生成を行うものです。

まず、"Linaqruf/anything-v3.0"という名前の事前学習済みモデルを使用してStableDiffusionPipelineオブジェクトを作成します。

次に、このモデルをCUDAに対応させます。

最後に、プロンプト "a girl is smiling"に対して画像を生成し、生成された画像を「smilegirl.png」という名前で保存します。

このプログラムの実行結果は以下です。