import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("borno1/stable_diffusion_2_base", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]5_number_model on Stable Diffusion via Dreambooth
model by borno1
This your the Stable Diffusion model fine-tuned the 5_number_model concept taught to Stable Diffusion with Dreambooth.
It can be used by modifying the instance_prompt: flower
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
Here are the images used for training this concept:
- Downloads last month
- 3