lykon-models/dreamshaper-6 is a Stable Diffusion model that has been fine-tuned on
runwayml/stable-diffusion-v1-5.
For more general information on how to run text-to-image models with 🧨 Diffusers, see
the docs.
1from diffusers import AutoPipelineForText2Image, DEISMultistepScheduler
2import torch
3
4pipe = AutoPipelineForText2Image.from_pretrained('lykon-models/dreamshaper-6', torch_dtype=torch.float16, variant="fp16")
5pipe.scheduler = DEISMultistepScheduler.from_config(pipe.scheduler.config)
6pipe = pipe.to("cuda")
7
8prompt = "portrait photo of muscular bearded guy in a worn mech suit, light bokeh, intricate, steel metal, elegant, sharp focus, soft lighting, vibrant colors"
9
10generator = torch.manual_seed(33)
11image = pipe(prompt, generator=generator, num_inference_steps=25).images[0]
12image.save("./image.png")