import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("Bugjuhjugjyy/bfb", dtype=torch.bfloat16, device_map="cuda")
prompt = "cat from bfb"
image = pipe(prompt).images[0]bfb Dreambooth model trained by Bugjuhjugjyy with TheLastBen's fast-DreamBooth notebook
Test the concept via A1111 Colab fast-Colab-A1111
cats are so good here:
this one's chill
him too 😎
why is this good? (prompt is cat from bfb for all)
- Downloads last month
- 12