How to use from the
Use from the
Diffusers library
pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline

# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("recoilme/ColorfulSSD-1B_v01", dtype=torch.bfloat16, device_map="cuda")

prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]

Text-to-image finetuning - recoilme/ColorfulSSD-1B_v01

This pipeline was finetuned from segmind/SSD-1B on the recoilme/aesthetic_photos_xs dataset. Below are some example images generated with the finetuned pipeline using the following prompt: a closeup of a beautiful woman with red hair and wearing blue and white striped shirt, 1girl, solo, looking_at_viewer, bangs, brown_hair, upper_body, kimono, freckles, realistic, red_lips:

img_0 img_1 img_2 img_3

Special VAE used for training: madebyollin/sdxl-vae-fp16-fix.

Downloads last month
2
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for recoilme/ColorfulSSD-1B_v01

Base model

segmind/SSD-1B
Finetuned
(6)
this model