import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("carlofkl/DreamLite-mobile", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]DreamLite-Mobile
Distilled single-branch variant of DreamLite-base for fast, on-device inference. No CFG; runs in ~4–8 steps at 1024×1024.
import torch
from diffusers import DreamLiteMobilePipeline
pipe = DreamLiteMobilePipeline.from_pretrained(
"carlofkl/DreamLite-mobile", torch_dtype=torch.bfloat16
).to("cuda")
image = pipe("a corgi astronaut", num_inference_steps=8).images[0]
License: CC BY-NC 4.0 (non-commercial). A full model card will be added once the diffusers integration PR is merged.
- Downloads last month
- 410
Model tree for carlofkl/DreamLite-mobile
Base model
carlofkl/DreamLite-base