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
- Draw Things
- DiffusionBee
Error when displaying image using pillow
#28
by jacob-joy - opened
UnidentifiedImageError Traceback (most recent call last)
in <cell line: 16>()
14 import io
15 from PIL import Image
---> 16 image = Image.open(io.BytesIO(image_bytes))
/usr/local/lib/python3.10/dist-packages/PIL/Image.py in open(fp, mode, formats)
3281 warnings.warn(message)
3282 msg = "cannot identify image file %r" % (filename if filename else fp)
-> 3283 raise UnidentifiedImageError(msg)
3284
3285
UnidentifiedImageError: cannot identify image file <_io.BytesIO object at 0x7b1d473d2390>
The Inference API is still not available since the internal diffusers version is not updated yet, please load the model and then try.