a573p/Face-Sketches40
Viewer • Updated • 40 • 71
How to use a573p/face-sketch-model-db with Diffusers:
pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("a573p/face-sketch-model-db", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]The Model face-sketch-model-db is a text-to-image diffusion model that has been finetuned with Dreambooth on 40 face sketches of the CUHK Face Sketch Database (CUFS). This model was trained as part of a university project, which should generate face sketch drawing more accurately. Base Model is runwayml/stable-diffusion-v1-5