├── .gitignore ├── LICENSE ├── README.md ├── assets └── sample.png ├── benchmark ├── andy_warhol_prompts.csv ├── art_prompts.csv ├── artwork_prompts.csv ├── big_artists_prompts.csv ├── caravaggio_prompts.csv ├── coco_30k.csv ├── famous_art_prompts.csv ├── generic_artists_prompts.csv ├── i2p_benchmark.csv ├── imagenet_prompts.csv ├── kelly_prompts.csv ├── niche_art_prompts.csv ├── nudity_benchmark.csv ├── picasso_prompts.csv ├── rembrandt_prompts.csv ├── short_niche_art_prompts.csv ├── short_vangogh_prompts.csv ├── small_imagenet_prompts.csv └── vangogh_prompts.csv ├── calculate_metrics.py ├── configs ├── generation.yaml ├── pikachu │ ├── config.yaml │ └── prompt.yaml └── snoopy │ ├── config.yaml │ └── prompt.yaml ├── demo.ipynb ├── evaluate_task.py ├── infer_spm.py ├── requirements.txt ├── src ├── __init__.py ├── configs │ ├── __init__.py │ ├── config.py │ ├── generation_config.py │ └── prompt.py ├── engine │ ├── __init__.py │ ├── sampling.py │ └── train_util.py ├── evaluation │ ├── __init__.py │ ├── artwork_evaluator.py │ ├── clip_evaluator.py │ ├── coco_evaluator.py │ ├── eval_util.py │ ├── evaluator.py │ └── i2p_evaluator.py ├── misc │ ├── __init__.py │ ├── clip_templates.py │ └── sld_pipeline.py └── models │ ├── __init__.py │ ├── merge_spm.py │ ├── model_util.py │ └── spm.py ├── tools ├── model_converters │ ├── convert_diffusers_to_original_stable_diffusion.py │ └── convert_original_stable_diffusion_to_diffusers.py ├── nearest_encoding.py └── nude_detection.py ├── train_spm.py ├── train_spm_xl.py └── train_spm_xl_mem_reduce.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/README.md -------------------------------------------------------------------------------- /assets/sample.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/assets/sample.png -------------------------------------------------------------------------------- /benchmark/andy_warhol_prompts.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/benchmark/andy_warhol_prompts.csv -------------------------------------------------------------------------------- /benchmark/art_prompts.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/benchmark/art_prompts.csv -------------------------------------------------------------------------------- /benchmark/artwork_prompts.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/benchmark/artwork_prompts.csv -------------------------------------------------------------------------------- /benchmark/big_artists_prompts.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/benchmark/big_artists_prompts.csv -------------------------------------------------------------------------------- /benchmark/caravaggio_prompts.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/benchmark/caravaggio_prompts.csv -------------------------------------------------------------------------------- /benchmark/coco_30k.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/benchmark/coco_30k.csv -------------------------------------------------------------------------------- /benchmark/famous_art_prompts.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/benchmark/famous_art_prompts.csv -------------------------------------------------------------------------------- /benchmark/generic_artists_prompts.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/benchmark/generic_artists_prompts.csv -------------------------------------------------------------------------------- /benchmark/i2p_benchmark.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/benchmark/i2p_benchmark.csv -------------------------------------------------------------------------------- /benchmark/imagenet_prompts.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/benchmark/imagenet_prompts.csv -------------------------------------------------------------------------------- /benchmark/kelly_prompts.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/benchmark/kelly_prompts.csv -------------------------------------------------------------------------------- /benchmark/niche_art_prompts.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/benchmark/niche_art_prompts.csv -------------------------------------------------------------------------------- /benchmark/nudity_benchmark.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/benchmark/nudity_benchmark.csv -------------------------------------------------------------------------------- /benchmark/picasso_prompts.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/benchmark/picasso_prompts.csv -------------------------------------------------------------------------------- /benchmark/rembrandt_prompts.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/benchmark/rembrandt_prompts.csv -------------------------------------------------------------------------------- /benchmark/short_niche_art_prompts.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/benchmark/short_niche_art_prompts.csv -------------------------------------------------------------------------------- /benchmark/short_vangogh_prompts.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/benchmark/short_vangogh_prompts.csv -------------------------------------------------------------------------------- /benchmark/small_imagenet_prompts.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/benchmark/small_imagenet_prompts.csv -------------------------------------------------------------------------------- /benchmark/vangogh_prompts.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/benchmark/vangogh_prompts.csv -------------------------------------------------------------------------------- /calculate_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/calculate_metrics.py -------------------------------------------------------------------------------- /configs/generation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/configs/generation.yaml -------------------------------------------------------------------------------- /configs/pikachu/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/configs/pikachu/config.yaml -------------------------------------------------------------------------------- /configs/pikachu/prompt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/configs/pikachu/prompt.yaml -------------------------------------------------------------------------------- /configs/snoopy/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/configs/snoopy/config.yaml -------------------------------------------------------------------------------- /configs/snoopy/prompt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/configs/snoopy/prompt.yaml -------------------------------------------------------------------------------- /demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/demo.ipynb -------------------------------------------------------------------------------- /evaluate_task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/evaluate_task.py -------------------------------------------------------------------------------- /infer_spm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/infer_spm.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/configs/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/src/configs/config.py -------------------------------------------------------------------------------- /src/configs/generation_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/src/configs/generation_config.py -------------------------------------------------------------------------------- /src/configs/prompt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/src/configs/prompt.py -------------------------------------------------------------------------------- /src/engine/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/engine/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/src/engine/sampling.py -------------------------------------------------------------------------------- /src/engine/train_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/src/engine/train_util.py -------------------------------------------------------------------------------- /src/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/src/evaluation/__init__.py -------------------------------------------------------------------------------- /src/evaluation/artwork_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/src/evaluation/artwork_evaluator.py -------------------------------------------------------------------------------- /src/evaluation/clip_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/src/evaluation/clip_evaluator.py -------------------------------------------------------------------------------- /src/evaluation/coco_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/src/evaluation/coco_evaluator.py -------------------------------------------------------------------------------- /src/evaluation/eval_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/src/evaluation/eval_util.py -------------------------------------------------------------------------------- /src/evaluation/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/src/evaluation/evaluator.py -------------------------------------------------------------------------------- /src/evaluation/i2p_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/src/evaluation/i2p_evaluator.py -------------------------------------------------------------------------------- /src/misc/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/misc/clip_templates.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/src/misc/clip_templates.py -------------------------------------------------------------------------------- /src/misc/sld_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/src/misc/sld_pipeline.py -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/merge_spm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/src/models/merge_spm.py -------------------------------------------------------------------------------- /src/models/model_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/src/models/model_util.py -------------------------------------------------------------------------------- /src/models/spm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/src/models/spm.py -------------------------------------------------------------------------------- /tools/model_converters/convert_diffusers_to_original_stable_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/tools/model_converters/convert_diffusers_to_original_stable_diffusion.py -------------------------------------------------------------------------------- /tools/model_converters/convert_original_stable_diffusion_to_diffusers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/tools/model_converters/convert_original_stable_diffusion_to_diffusers.py -------------------------------------------------------------------------------- /tools/nearest_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/tools/nearest_encoding.py -------------------------------------------------------------------------------- /tools/nude_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/tools/nude_detection.py -------------------------------------------------------------------------------- /train_spm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/train_spm.py -------------------------------------------------------------------------------- /train_spm_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/train_spm_xl.py -------------------------------------------------------------------------------- /train_spm_xl_mem_reduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Con6924/SPM/HEAD/train_spm_xl_mem_reduce.py --------------------------------------------------------------------------------