├── README.md ├── models ├── autoencoder_kl.py ├── modules.py ├── transformer_sd3_cnext.py └── transformer_sd3_cnext_ip.py ├── pipeline.png ├── pipelines ├── pipeline_stable_diffusion_3_cnext.py ├── pipeline_stable_diffusion_3_cnext_sw.py └── pipeline_stable_diffusion_3_cnext_sw_shift.py ├── requirements.txt ├── teaser.png ├── test_cnext_sd3_sw_image_prompt.py └── train_cnext_sd3_image_prompt.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongdehong/DPIR/HEAD/README.md -------------------------------------------------------------------------------- /models/autoencoder_kl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongdehong/DPIR/HEAD/models/autoencoder_kl.py -------------------------------------------------------------------------------- /models/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongdehong/DPIR/HEAD/models/modules.py -------------------------------------------------------------------------------- /models/transformer_sd3_cnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongdehong/DPIR/HEAD/models/transformer_sd3_cnext.py -------------------------------------------------------------------------------- /models/transformer_sd3_cnext_ip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongdehong/DPIR/HEAD/models/transformer_sd3_cnext_ip.py -------------------------------------------------------------------------------- /pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongdehong/DPIR/HEAD/pipeline.png -------------------------------------------------------------------------------- /pipelines/pipeline_stable_diffusion_3_cnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongdehong/DPIR/HEAD/pipelines/pipeline_stable_diffusion_3_cnext.py -------------------------------------------------------------------------------- /pipelines/pipeline_stable_diffusion_3_cnext_sw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongdehong/DPIR/HEAD/pipelines/pipeline_stable_diffusion_3_cnext_sw.py -------------------------------------------------------------------------------- /pipelines/pipeline_stable_diffusion_3_cnext_sw_shift.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongdehong/DPIR/HEAD/pipelines/pipeline_stable_diffusion_3_cnext_sw_shift.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongdehong/DPIR/HEAD/requirements.txt -------------------------------------------------------------------------------- /teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongdehong/DPIR/HEAD/teaser.png -------------------------------------------------------------------------------- /test_cnext_sd3_sw_image_prompt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongdehong/DPIR/HEAD/test_cnext_sd3_sw_image_prompt.py -------------------------------------------------------------------------------- /train_cnext_sd3_image_prompt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongdehong/DPIR/HEAD/train_cnext_sd3_image_prompt.py --------------------------------------------------------------------------------