├── .github └── CODEOWNERS ├── .gitignore ├── LICENSE ├── MODEL_CARD.md ├── README.md ├── data ├── laion_aes │ └── .gitkeep ├── mscoco_val2014_30k │ └── .gitkeep └── mscoco_val2014_41k_full │ └── .gitkeep ├── requirements.txt ├── results └── .gitkeep ├── scripts ├── eval_scores.sh ├── finetune_full.sh ├── finetune_lora.sh ├── generate_after_full_ft.sh ├── generate_after_lora_ft.sh ├── generate_with_trained_unet.sh ├── get_laion_data.sh ├── get_mscoco_files.sh ├── kd_train.sh ├── kd_train_2m.sh ├── kd_train_toy.sh ├── kd_train_toy_ddp.sh ├── kd_train_v2-base-im512.sh └── kd_train_v2-im768.sh └── src ├── __init__.py ├── count_macs_params.py ├── dreambooth_finetune.py ├── eval_clip_score.py ├── generate.py ├── generate_single_prompt.py ├── get_stat_mscoco_val2014.py ├── kd_train_text_to_image.py ├── unet_config ├── bk_base │ └── config.json ├── bk_small │ └── config.json └── bk_tiny │ └── config.json ├── unet_config_v2-base ├── bk_base │ └── config.json ├── bk_small │ └── config.json └── bk_tiny │ └── config.json ├── unet_config_v2 ├── bk_base │ └── config.json ├── bk_small │ └── config.json └── bk_tiny │ └── config.json └── utils ├── __init__.py ├── inference_pipeline.py └── misc.py /.github/CODEOWNERS: -------------------------------------------------------------------------------- 1 | * @bokyeong1015 -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/LICENSE -------------------------------------------------------------------------------- /MODEL_CARD.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/MODEL_CARD.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/README.md -------------------------------------------------------------------------------- /data/laion_aes/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/mscoco_val2014_30k/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/mscoco_val2014_41k_full/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/requirements.txt -------------------------------------------------------------------------------- /results/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /scripts/eval_scores.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/scripts/eval_scores.sh -------------------------------------------------------------------------------- /scripts/finetune_full.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/scripts/finetune_full.sh -------------------------------------------------------------------------------- /scripts/finetune_lora.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/scripts/finetune_lora.sh -------------------------------------------------------------------------------- /scripts/generate_after_full_ft.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/scripts/generate_after_full_ft.sh -------------------------------------------------------------------------------- /scripts/generate_after_lora_ft.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/scripts/generate_after_lora_ft.sh -------------------------------------------------------------------------------- /scripts/generate_with_trained_unet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/scripts/generate_with_trained_unet.sh -------------------------------------------------------------------------------- /scripts/get_laion_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/scripts/get_laion_data.sh -------------------------------------------------------------------------------- /scripts/get_mscoco_files.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/scripts/get_mscoco_files.sh -------------------------------------------------------------------------------- /scripts/kd_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/scripts/kd_train.sh -------------------------------------------------------------------------------- /scripts/kd_train_2m.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/scripts/kd_train_2m.sh -------------------------------------------------------------------------------- /scripts/kd_train_toy.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/scripts/kd_train_toy.sh -------------------------------------------------------------------------------- /scripts/kd_train_toy_ddp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/scripts/kd_train_toy_ddp.sh -------------------------------------------------------------------------------- /scripts/kd_train_v2-base-im512.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/scripts/kd_train_v2-base-im512.sh -------------------------------------------------------------------------------- /scripts/kd_train_v2-im768.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/scripts/kd_train_v2-im768.sh -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/count_macs_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/src/count_macs_params.py -------------------------------------------------------------------------------- /src/dreambooth_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/src/dreambooth_finetune.py -------------------------------------------------------------------------------- /src/eval_clip_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/src/eval_clip_score.py -------------------------------------------------------------------------------- /src/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/src/generate.py -------------------------------------------------------------------------------- /src/generate_single_prompt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/src/generate_single_prompt.py -------------------------------------------------------------------------------- /src/get_stat_mscoco_val2014.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/src/get_stat_mscoco_val2014.py -------------------------------------------------------------------------------- /src/kd_train_text_to_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/src/kd_train_text_to_image.py -------------------------------------------------------------------------------- /src/unet_config/bk_base/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/src/unet_config/bk_base/config.json -------------------------------------------------------------------------------- /src/unet_config/bk_small/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/src/unet_config/bk_small/config.json -------------------------------------------------------------------------------- /src/unet_config/bk_tiny/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/src/unet_config/bk_tiny/config.json -------------------------------------------------------------------------------- /src/unet_config_v2-base/bk_base/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/src/unet_config_v2-base/bk_base/config.json -------------------------------------------------------------------------------- /src/unet_config_v2-base/bk_small/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/src/unet_config_v2-base/bk_small/config.json -------------------------------------------------------------------------------- /src/unet_config_v2-base/bk_tiny/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/src/unet_config_v2-base/bk_tiny/config.json -------------------------------------------------------------------------------- /src/unet_config_v2/bk_base/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/src/unet_config_v2/bk_base/config.json -------------------------------------------------------------------------------- /src/unet_config_v2/bk_small/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/src/unet_config_v2/bk_small/config.json -------------------------------------------------------------------------------- /src/unet_config_v2/bk_tiny/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/src/unet_config_v2/bk_tiny/config.json -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/utils/inference_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/src/utils/inference_pipeline.py -------------------------------------------------------------------------------- /src/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Nota-NetsPresso/BK-SDM/HEAD/src/utils/misc.py --------------------------------------------------------------------------------