Views
No views yet

pip install diffusers torch accelerate transformers1from diffusers import DiffusionPipeline
2import torch
3
4pipe = DiffusionPipeline.from_pretrained("playgroundai/playground-v1")
5
6if torch.cuda.is_available():
7 pipe.to("cuda")
8
9prompt = "An Astronaut in a jungle, photorealistic"
10image = pipe(prompt=prompt).images[0]
11
12image.save("playground-v1.png")