This is a LoRA in ComfyUI format for Flux.2 Klein 9B trained to generate images of the person who appears in the reference image using a facial recognition loss along with a teacher distillation loss that keeps it in line with the distilled model's outputs. Trained on UTKFace and a subset of VGGFace2 images.

The original distilled model does this pretty well, but often generates a similar-but-different person.

Here are a few examples (Facial Recognition Score from CVLFace-ViT-webface4m. Higher is better)

All generations with seed 432262096973502

Input Distilled Distilled Score LoRA LoRA Score
test1 Gen1_Distilled .4372 Gen2_Lora .4451
testsame2 Gen4_Distilled .5373 Gen4_Lora .5524
Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support