from diffusers import StableDiffusionXLPipeline
import torch

pipeline = StableDiffusionXLPipeline.from_pretrained(
    "BraceZHY/sdxl-DPO100K",
    torch_dtype=torch.float16,
    use_safetensors=True
)

pipeline = pipeline.to("cuda")

prompt = "Your prompt here"
image = pipeline(prompt).images[0]
image.save("output.png")

Training Details

Base Model: stabilityai/stable-diffusion-xl-base-1.0
LoRA Weights: Merged from ./sdxl-DPO100K
Model Type: Stable Diffusion XL
Format: Diffusers

This model was created by training LoRA weights and then merging them with the base model for easier deployment and faster inference.