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] 'list' object cannot be interpreted as an integer
#31
by cerizzi - opened
I'm trying to compile/convert the model for using it with AWS Neuron. Running: "optimum-cli export neuron --model segmind/SSD-1B --batch_size 1 --height 1024 --width 1024 --auto_cast matmul --auto_cast_type bf16 sdxl_segmind_neuron/" I get the following python error: "'list' object cannot be interpreted as an integer". I thought that was an incompatibility problem with AWS Neuron, but I get the exact same error even if I try the inference using HF inference. I'm attaching a screenshot.
The Inference API is not yet supported please use the Demo Space instead.
Ok, thank you so much! So, I assume also: https://huggingface.co/docs/optimum-neuron/main/en/package_reference/export is not supported.
