import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("RobertLau/cat-toy", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]Cat toy on Stable Diffusion via Dreambooth
model by RobertLau
This your the Stable Diffusion model fine-tuned the Cat toy concept taught to Stable Diffusion with Dreambooth.
It can be used by modifying the instance_prompt: toy
You can also train your own concepts and upload them to the library by using this notebook.
And you can run your new concept via diffusers: Colab Notebook for Inference, Spaces with the Public Concepts loaded
Usage:
If you want to use this concept, please add <cat-toy> in your prompt, for example: "A <cat-toy> in mad max fury road"
Here are the images used for training this concept:
- Downloads last month
- 1