Views
No views yet
1from diffusers import StableDiffusionPipeline
2import torch
3
4pipeline = StableDiffusionPipeline.from_pretrained(
5 "untitled . pony",
6 torch_dtype=torch.float16,
7 safety_checker=None
8)
9if torch.cuda.is_available():
10 pipeline = pipeline.to("cuda")
11
12image = pipeline(
13 prompt="your prompt here",
14 num_inference_steps=30,
15 guidance_scale=7.0
16).images[0]
17image.save("output.png")