tools / run_xl_lora.py
patrickvonplaten's picture
up
71da11e
raw
history blame
340 Bytes
#!/usr/bin/env python3
from diffusers import DiffusionPipeline
import torch
pipe = DiffusionPipeline.from_pretrained("/home/patrick/sai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16)
pipe.load_lora_weights("/home/patrick/sai/stable-diffusion-xl-base-1.0/sd_xl_offset_example-lora_1.0.safetensors")
import ipdb; ipdb.set_trace()