├── README.md ├── assets ├── abstract_x1.png ├── aes_score_rank.png ├── checkpoints.png ├── clip_score_rank.png ├── compare.png ├── image-2.png ├── image-3.png ├── image-4.png └── image.png ├── preprocess_dataset.py ├── scorer ├── README.md ├── __init__.py ├── aesthetic_scorer.py ├── base_scorer.py ├── clip_scorer.py ├── simulacra_aesthetic_models │ ├── __init_.py │ ├── rank_images.py │ ├── simulacra_compute_embeddings.py │ └── simulacra_fit_linear_model.py └── simulacra_aesthetic_scorer.py ├── scripts ├── preprocess_dataset.sh └── train.sh └── train.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qqingzheng/AI-Self-Training-DPO-SDXL/HEAD/README.md -------------------------------------------------------------------------------- /assets/abstract_x1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qqingzheng/AI-Self-Training-DPO-SDXL/HEAD/assets/abstract_x1.png -------------------------------------------------------------------------------- /assets/aes_score_rank.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qqingzheng/AI-Self-Training-DPO-SDXL/HEAD/assets/aes_score_rank.png -------------------------------------------------------------------------------- /assets/checkpoints.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qqingzheng/AI-Self-Training-DPO-SDXL/HEAD/assets/checkpoints.png -------------------------------------------------------------------------------- /assets/clip_score_rank.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qqingzheng/AI-Self-Training-DPO-SDXL/HEAD/assets/clip_score_rank.png -------------------------------------------------------------------------------- /assets/compare.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qqingzheng/AI-Self-Training-DPO-SDXL/HEAD/assets/compare.png -------------------------------------------------------------------------------- /assets/image-2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qqingzheng/AI-Self-Training-DPO-SDXL/HEAD/assets/image-2.png -------------------------------------------------------------------------------- /assets/image-3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qqingzheng/AI-Self-Training-DPO-SDXL/HEAD/assets/image-3.png -------------------------------------------------------------------------------- /assets/image-4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qqingzheng/AI-Self-Training-DPO-SDXL/HEAD/assets/image-4.png -------------------------------------------------------------------------------- /assets/image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qqingzheng/AI-Self-Training-DPO-SDXL/HEAD/assets/image.png -------------------------------------------------------------------------------- /preprocess_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qqingzheng/AI-Self-Training-DPO-SDXL/HEAD/preprocess_dataset.py -------------------------------------------------------------------------------- /scorer/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qqingzheng/AI-Self-Training-DPO-SDXL/HEAD/scorer/README.md -------------------------------------------------------------------------------- /scorer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qqingzheng/AI-Self-Training-DPO-SDXL/HEAD/scorer/__init__.py -------------------------------------------------------------------------------- /scorer/aesthetic_scorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qqingzheng/AI-Self-Training-DPO-SDXL/HEAD/scorer/aesthetic_scorer.py -------------------------------------------------------------------------------- /scorer/base_scorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qqingzheng/AI-Self-Training-DPO-SDXL/HEAD/scorer/base_scorer.py -------------------------------------------------------------------------------- /scorer/clip_scorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qqingzheng/AI-Self-Training-DPO-SDXL/HEAD/scorer/clip_scorer.py -------------------------------------------------------------------------------- /scorer/simulacra_aesthetic_models/__init_.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /scorer/simulacra_aesthetic_models/rank_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qqingzheng/AI-Self-Training-DPO-SDXL/HEAD/scorer/simulacra_aesthetic_models/rank_images.py -------------------------------------------------------------------------------- /scorer/simulacra_aesthetic_models/simulacra_compute_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qqingzheng/AI-Self-Training-DPO-SDXL/HEAD/scorer/simulacra_aesthetic_models/simulacra_compute_embeddings.py -------------------------------------------------------------------------------- /scorer/simulacra_aesthetic_models/simulacra_fit_linear_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qqingzheng/AI-Self-Training-DPO-SDXL/HEAD/scorer/simulacra_aesthetic_models/simulacra_fit_linear_model.py -------------------------------------------------------------------------------- /scorer/simulacra_aesthetic_scorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qqingzheng/AI-Self-Training-DPO-SDXL/HEAD/scorer/simulacra_aesthetic_scorer.py -------------------------------------------------------------------------------- /scripts/preprocess_dataset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qqingzheng/AI-Self-Training-DPO-SDXL/HEAD/scripts/preprocess_dataset.sh -------------------------------------------------------------------------------- /scripts/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qqingzheng/AI-Self-Training-DPO-SDXL/HEAD/scripts/train.sh -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qqingzheng/AI-Self-Training-DPO-SDXL/HEAD/train.py --------------------------------------------------------------------------------