Diffusersのset_adapters()を使用してLoRAの重みを設定する
Table of Contents
LoRAの重みを設定する
set_adapters()を使用してLoRAの重みを設定し、モデルに対するLoRAの影響を調整します。
import torch
from diffusers import StableDiffusionPipeline
model_id = "./my_model.safetensors"
seed = 0123
# モデルを読み込む
pipe = StableDiffusionPipeline.from_single_file(
model_id,
torch_dtype=torch.float16)
pipe.to("cuda")
lora_id = "./my_lora.safetensors"
adapter_name = "my_lora"
adapter_weight = 0.5
# LoRAを読み込む
pipe.load_lora_weights(lora_path, adapter_name=adapter_name)
# LoRAの重みを[0., 1.]の範囲で設定
pipe.set_adapters(adapter_name, adapter_weight)
prompt = "a prompt for the model"
image = pipe(
prompt,
generator = torch.Generator(device="cuda").manual_seed(seed)
).images[0]