Views
No views yet
1import torch
2from diffusers import StableDiffusionPipeline
3
4model_id = "SorianoTech/pontia-generador-imagenes-vintage"
5device = "cuda"
6
7
8pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
9pipe = pipe.to(device)
10
11prompt = "a photo of an astronaut riding a horse on mars"
12image = pipe(prompt).images[0]
13
14image.save("astronaut_rides_horse.png")