qiacheng's picture
Update README.md
334cf6a
|
raw
history blame contribute delete
No virus
666 Bytes
metadata
license: apache-2.0

SDXL-base-1.0 LDM: https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0

SDXL-base-1.0 LCM-Lora: https://huggingface.co/latent-consistency/lcm-lora-sdxl

SDXL model fused with LCM Lora, saved using diffusers .save_pretrained()

Sample usage:

import torch
from diffusers import StableDiffusionXLPipeline

pipe = StableDiffusionXLPipeline.from_pretrained("qiacheng/stable-diffusion-xl-base-1.0-lcm")

prompt = "a cat"

height = 1024
width = 1024
steps = 6
guidance_scale = 1

output = pipe(prompt=prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=guidance_scale, output_type="pil")