Views
No views yet
1from diffusers import StableDiffusionPipeline
2
3model_path = "johnowhitaker/rainbowdiffusion"
4pipe = StableDiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.float16)
5pipe.to("cuda")
6
7image = pipe(prompt="A cat").images[0]
8image