import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("prajwal13/pintu_dreambooth", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]Model Card for Dreambooth model trained on My pet Pintu's images
This model is a diffusion model for unconditional image generation of my cute pet dog Pintu trained using Dreambooth concept. The token to use is sks .
Usage
from diffusers import StableDiffusionPipeline
pipeline = StableDiffusionPipeline.from_pretrained('prajwal13/pintu_dreambooth')
image = pipeline('a photo of sks dog').images[0]
image
These are the images on which the dreambooth model is trained on
- Downloads last month
- 14