import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("axolotron/ice-cream-animals", dtype=torch.bfloat16, device_map="cuda")
prompt = "a butterfly ice cream, icenimal"
image = pipe(prompt).images[0]Ice_cream_animals Dreambooth Model for Food trained on a custom dataset.
This is a Stable Diffusion 2.1 768px model fine-tuned on the food concept with DreamBooth. It can be used by modifying the instance_prompt: a butterfly ice cream, icenimal
This model was created as part of the DreamBooth Hackathon 🔥.
Samples:
A red dragon
A disney princess
A demogorgon
An elephant
A bee
An axolotl
a cat
Pokemon
Donald Trump as ice cream
A butterfly
- Downloads last month
- 23