├── .gitignore ├── .vscode └── launch.json ├── README.md ├── in ├── gt1.png ├── gt2.png └── gt3.png ├── requirements.txt ├── script ├── check_wds_for_nan.py ├── inference_decoder.py └── make_sdxl_latent_dataset.py └── sdxl_diff_dec ├── dataset └── folder_of_images.py ├── normalize.py ├── prune_conv_out.py ├── sampling.py ├── schedule.py ├── sd_denoiser.py └── vae └── attn ├── natten_attn_processor.py ├── null_attn_processor.py └── qkv_fusion.py /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__/ 2 | /out/ 3 | /in/ -------------------------------------------------------------------------------- /.vscode/launch.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Birch-san/sdxl-diffusion-decoder/HEAD/.vscode/launch.json -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /in/gt1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Birch-san/sdxl-diffusion-decoder/HEAD/in/gt1.png -------------------------------------------------------------------------------- /in/gt2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Birch-san/sdxl-diffusion-decoder/HEAD/in/gt2.png -------------------------------------------------------------------------------- /in/gt3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Birch-san/sdxl-diffusion-decoder/HEAD/in/gt3.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Birch-san/sdxl-diffusion-decoder/HEAD/requirements.txt -------------------------------------------------------------------------------- /script/check_wds_for_nan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Birch-san/sdxl-diffusion-decoder/HEAD/script/check_wds_for_nan.py -------------------------------------------------------------------------------- /script/inference_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Birch-san/sdxl-diffusion-decoder/HEAD/script/inference_decoder.py -------------------------------------------------------------------------------- /script/make_sdxl_latent_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Birch-san/sdxl-diffusion-decoder/HEAD/script/make_sdxl_latent_dataset.py -------------------------------------------------------------------------------- /sdxl_diff_dec/dataset/folder_of_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Birch-san/sdxl-diffusion-decoder/HEAD/sdxl_diff_dec/dataset/folder_of_images.py -------------------------------------------------------------------------------- /sdxl_diff_dec/normalize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Birch-san/sdxl-diffusion-decoder/HEAD/sdxl_diff_dec/normalize.py -------------------------------------------------------------------------------- /sdxl_diff_dec/prune_conv_out.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Birch-san/sdxl-diffusion-decoder/HEAD/sdxl_diff_dec/prune_conv_out.py -------------------------------------------------------------------------------- /sdxl_diff_dec/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Birch-san/sdxl-diffusion-decoder/HEAD/sdxl_diff_dec/sampling.py -------------------------------------------------------------------------------- /sdxl_diff_dec/schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Birch-san/sdxl-diffusion-decoder/HEAD/sdxl_diff_dec/schedule.py -------------------------------------------------------------------------------- /sdxl_diff_dec/sd_denoiser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Birch-san/sdxl-diffusion-decoder/HEAD/sdxl_diff_dec/sd_denoiser.py -------------------------------------------------------------------------------- /sdxl_diff_dec/vae/attn/natten_attn_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Birch-san/sdxl-diffusion-decoder/HEAD/sdxl_diff_dec/vae/attn/natten_attn_processor.py -------------------------------------------------------------------------------- /sdxl_diff_dec/vae/attn/null_attn_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Birch-san/sdxl-diffusion-decoder/HEAD/sdxl_diff_dec/vae/attn/null_attn_processor.py -------------------------------------------------------------------------------- /sdxl_diff_dec/vae/attn/qkv_fusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Birch-san/sdxl-diffusion-decoder/HEAD/sdxl_diff_dec/vae/attn/qkv_fusion.py --------------------------------------------------------------------------------