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]
OSError: /data/models/SSD-1B does not appear to have a file named config.json. Checkout 'https://huggingface.co//data/models/SSD-1B/main' for available files
thanks a lot
transformers==4.34.1
· Sign up or log in to comment
