import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("segmind/SSD-1B", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("openskyml/lcm-lora-ssd-1b")
prompt = "A cute tiny mouse holding a piece of luxury cheese, pixar ratatouille"
image = pipe(prompt).images[0]LCM LoRA SSD-1B
- Prompt
- A cute tiny mouse holding a piece of luxury cheese, pixar ratatouille
- Negative Prompt
- text, blurrym ugly
Trigger words
You should use lora:pytorch_lora_weights:0.3 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
- 11
Model tree for openskyml/lcm-lora-ssd-1b
Base model
segmind/SSD-1B