Diffusersのfrom_single_file()で単一のDiffusionモデルファイルをロードする
Table of Contents
Diffusersのfrom_single_file()
from_pretrained()の代わりにfrom_single_file()を使用して、単一ファイルからDiffusionモデルを読み込みます。
import torch
from diffusers import StableDiffusionPipeline
model_id = "./my-model.safetensors"
pipe = StableDiffusionPipeline.from_single_file(
model_id,
torch_dtype=torch.float16)
pipe.to("cuda")
prompt = "a prompt for the model"
image = pipe(prompt).images[0]