Face0 - SDXL

Unofficial implementation of Face0 with SDXL

Train

accelerate launch \
    --multi_gpu \
    --num_processes 4 \
    train.py train_from_yaml \
    --config_path=configs/train.yaml

Inference

• See notebooks/inference.ipynb

TODO

• Train with LoRA or Adapter
• Apply EMA
• Implement guidance method described in section 2.2
• Add a legit inference script

Credits

• Training code based on diffusers🧨
• Inference pipeline from kohya_ss
• Face encoder from facenet-pytorch