README.md

metadata

license: apache-2.0
datasets:
  - Rapidata/Face_Generation_Benchmark
language:
  - en
base_model:
  - stabilityai/stable-diffusion-xl-base-1.0
pipeline_tag: text-to-image
library_name: diffusers
tags:
  - text-to-image
  - lora
  - diffusers

Model description

Portrait-generator

Portrait-generator is a LoRA fine-tuned adapter for the stable-diffusion-xl-base-1.0 model, trained on the Rapidata/Face_Generation_Benchmark dataset. It specializes in generating face images.

---

Usage

1. Install dependencies (if not installed):

pip install diffusers transformers accelerate safetensor torch

2. Load stable-diffusion-xl-base-1.0 and LoRA adapter:

from diffusers import DiffusionPipeline
import torch

pipe = DiffusionPipeline.from_pretrained(
    "stabilityai/stable-diffusion-xl-base-1.0",
    torch_dtype=torch.float16,
).to("cuda")

pipe.load_lora_weights("SkyAsl/Portrait-generator")
pipe.fuse_lora()

image = pipe("a high-quality portrait of a woman").images[0]
image.save("test.png")

---

Metrics

• Average CLIPScore: 16.6980

---