File size: 2,922 Bytes
8574665 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 | ---
license: other
base_model: black-forest-labs/FLUX.2-klein-base-4B
tags:
- face-restoration
- blind-face-restoration
- reference-guided-restoration
- flux
- lora
- computer-vision
pipeline_tag: image-to-image
---
# IConFace LoRA Adapter
This repository hosts the released LoRA adapter for **IConFace: Identity-Structure
Asymmetric Conditioning for Unified Reference-Aware Face Restoration**.
- Project page: https://cosmicrealm.github.io/IConFace/
- arXiv: https://arxiv.org/abs/2605.02814
- Code: https://github.com/cosmicrealm/IConFace
- Model weights: https://huggingface.co/zhangjinyang/IConFace
## Visual Results
<p align="center">
<img src="assets/teaser_example.png" alt="IConFace detail restoration example" width="900">
</p>
### Reference-Aware Restoration
<p align="center">
<img src="assets/ref-aware_vis.png" alt="Reference-aware visual results" width="900">
</p>
### No-Reference Restoration
<p align="center">
<img src="assets/noref-aware_vis.png" alt="No-reference visual results" width="900">
</p>
More visual comparisons are available on the project page.
## Files
```text
iconface_lora.safetensors
```
This adapter is used with:
- Base model: `black-forest-labs/FLUX.2-klein-base-4B`
- LoRA target preset: `flux2_klein`
- LoRA rank: 16
- Expected path in the code release: `pretrained_models/iconface_lora.safetensors`
## Usage
Use the GitHub code release:
```bash
git clone https://github.com/cosmicrealm/IConFace.git
cd IConFace
pip install -r requirements.txt
bash scripts/download_models.sh
```
No-reference inference:
```bash
python scripts/infer_no_ref.py \
--model_dir pretrained_models/FLUX.2-klein-base-4B \
--lora_path pretrained_models/iconface_lora.safetensors \
--adaface_model_path pretrained_models/adaface_ir50_ms1mv2.ckpt \
--condition_image examples/lq.png \
--output_dir outputs/no_ref_demo
```
Reference-aware inference:
```bash
python scripts/infer_with_ref.py \
--model_dir pretrained_models/FLUX.2-klein-base-4B \
--lora_path pretrained_models/iconface_lora.safetensors \
--adaface_model_path pretrained_models/adaface_ir50_ms1mv2.ckpt \
--condition_image examples/lq.png \
--reference_image examples/ref1.png \
--reference_image examples/ref2.png \
--reference_image examples/ref3.png \
--output_dir outputs/ref_demo
```
## Citation
```bibtex
@misc{niu2026iconface,
title={IConFace: Identity-Structure Asymmetric Conditioning for Unified Reference-Aware Face Restoration},
author={Niu, Axi and Zhang, Jinyang and Qing, Senyan},
year={2026},
eprint={2605.02814},
archivePrefix={arXiv},
primaryClass={cs.CV},
doi={10.48550/arXiv.2605.02814},
url={https://arxiv.org/abs/2605.02814}
}
```
## License Notes
This adapter depends on the FLUX.2-Klein-base-4B base model. Users must comply
with the license and access terms of the base model and all external identity
models used by the code release.
|