Views
No views yet
1from diffusers import DiffusionPipeline
2import torch
3
4pipe = DiffusionPipeline.from_pretrained("freshcodestech/cat_in_space", torch_dtype=torch.float16)
5pipe.to("cuda")
6
7image = pipe("a cartoon cat in a space suit floating in a colorful galaxy").images[0]
8image.save("cat.png")