Ryan-sjtu/celebahq-caption
Viewer • Updated • 30k • 63 • 7
How to use lokesh6309/stablediff_celeba with Diffusers:
pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("lokesh6309/stablediff_celeba")
prompt = "A young woman with brown hair, a white hair accessory, and brown eyes. She has a neutral expression and appears to be in her 20s."
image = pipe(prompt).images[0]
You should use face to trigger the image generation.
Weights for this model are available in Safetensors format.
Download them in the Files & versions tab.
Base model
stable-diffusion-v1-5/stable-diffusion-v1-5