muneebable/imran-khan
Viewer • Updated • 17 • 70
How to use muneebable/imran-khan-dreambooth with Diffusers:
pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("muneebable/imran-khan-dreambooth", dtype=torch.bfloat16, device_map="cuda")
prompt = "imran-khan-804 person as a super hero hulk,"
image = pipe(prompt).images[0]import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("muneebable/imran-khan-dreambooth", dtype=torch.bfloat16, device_map="cuda")
prompt = "imran-khan-804 person as a super hero hulk,"
image = pipe(prompt).images[0]This is a Stable Diffusion model fine-tuned on the imran-khan-804 person concept with DreamBooth. It can be used by modifying the instance_prompt: imran-khan-804 person
This model was created as part of the DreamBooth Hackathon 🔥. Visit the organisation page for instructions on how to take part!
This is a Stable Diffusion model fine-tuned on person images for the wildcard theme.
from diffusers import StableDiffusionPipeline
pipeline = StableDiffusionPipeline.from_pretrained('muneebable/imran-khan-dreambooth')
image = pipeline().images[0]
image
The Training Notebook is available here: https://www.kaggle.com/code/muneeb2405/dreambooth-tranining
Here are the images used for training this concept:
Base model
stabilityai/stable-diffusion-2