import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("Kugos/KgSelfie_lr_15e-6", 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(Kugos/KgSelfie_lr_15e-6) 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
- 7