Views
No views yet
1import torch
2from diffusers import StableDiffusionPipeline
3
4pipe = StableDiffusionPipeline.from_pretrained("qiacheng/stable-diffusion-v1-5-lcm")
5
6prompt = "a cat"
7
8height = 512
9width = 512
10steps = 6
11guidance_scale = 1
12
13output = pipe(prompt=prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=guidance_scale, output_type="pil")
14