import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("johnowhitaker/lora-sdxl-plushie")
prompt = "a sbu dog on a bed"
image = pipe(prompt).images[0]SDXL LoRA DreamBooth - lora-sdxl-plushie
- Prompt
- a sbu dog on a bed
- Prompt
- a sbu dog on a bed
- Prompt
- a sbu dog on a bed
- Prompt
- a sbu dog on a bed
Model description
These are lora-sdxl-plushie LoRA adaption weights for stabilityai/stable-diffusion-xl-base-1.0.
The weights were trained using DreamBooth.
LoRA for the text encoder was enabled: False.
Special VAE used for training: None.
Trigger words
You should use plushie to trigger the image generation.
Download model
Weights for this model are available in Safetensors format.
Download them in the Files & versions tab.
- Downloads last month
- 166
Model tree for johnowhitaker/lora-sdxl-plushie
Base model
stabilityai/stable-diffusion-xl-base-1.0