pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("Robb753/robbin", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]robbin Dreambooth model trained by Robb753 with buildspace's DreamBooth notebook
Build your own using the AI Avatar project!
To get started head over to the project dashboard.
Sample pictures of this concept:
- Downloads last month
- -
# Gated model: Login with a HF token with gated access permission hf auth login