Text-to-Image
Diffusers
Safetensors
StableDiffusionXLPipeline
ultra-realistic
stable-diffusion
distilled-model
knowledge-distillation
Instructions to use segmind/SSD-1B with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use segmind/SSD-1B with Diffusers:
pip install -U diffusers transformers accelerate
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") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- Draw Things
- DiffusionBee
where does it save the generated image
#43
by Groows - opened
This comment has been hidden (marked as Resolved)
This comment has been hidden (marked as Resolved)
Groows changed discussion title from ImportError: Using `low_cpu_mem_usage=True`, a `device_map` or a `tp_plan` requires Accelerate: `pip install 'accelerate>=0.26.0'` to where does it save the generated image