Downloads · 30 days
0
0% of all-time downloads
Vargol/lcm-ssd-1b-full-model
lcm-ssd-1b-full-model is a text-to-image model from Vargol. Use it when you need an image from a text prompt. It is set up for diffusers. The card lists the license as openrail++.
This is a copy of the SSD-1B model (https://huggingface.co/segmind/SSD-1B) with the unet replaced with the LCM distilled unet (https://huggingface.co/latent-consistency/lcm-ssd-1b) and scheduler config set to default…
Downloads · 30 days
0
0% of all-time downloads
All-time downloads
441
Public
Parameters
1.3B
13.4 GB on disk
Likes
1
Public
Click a slice to open those files.
.safetensors13.4 GB · 100%
From the Hugging Face model README
This is a copy of the SSD-1B model (https://huggingface.co/segmind/SSD-1B) with the unet replaced with the LCM distilled unet (https://huggingface.co/latent-consistency/lcm-ssd-1b) and scheduler config set to default to the LCM Scheduler.
This makes LCM SSD-1B run as a standard Diffusion Pipeline
from diffusers import DiffusionPipeline
import torch
pipe = DiffusionPipeline.from_pretrained(
"Vargol/lcm-ssd-1b-full-model", variant='fp16', torch_dtype=torch.float16
).to("mps")
prompt = "Self-portrait oil painting, a beautiful cyborg with golden hair, 8k"
generator = torch.manual_seed(0)
image = pipe(
prompt=prompt, num_inference_steps=4, generator=generator, guidance_scale=8.0
).images[0]
image.save('distilled.png')