import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("freshcodes/cat_in_space")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]Cat in Space 🐱🚀
This is a Stable Diffusion XL model fine-tuned to generate cartoon cats in space.
Example
from diffusers import DiffusionPipeline
import torch
pipe = DiffusionPipeline.from_pretrained("freshcodestech/cat_in_space", torch_dtype=torch.float16)
pipe.to("cuda")
image = pipe("a cartoon cat in a space suit floating in a colorful galaxy").images[0]
image.save("cat.png")
- Downloads last month
- 26
Model tree for freshcodes/cat_in_space
Base model
stabilityai/stable-diffusion-xl-base-1.0