├── .gitignore ├── README.md ├── asset ├── dataset.jpg ├── demo.jpg └── sc-vae.png ├── autoencoder_kl.py ├── ds_config.yaml ├── eval.py ├── generate_and_eval_fid.py ├── improved-aesthetic-predictor └── sac+logos+ava1-l14-linearMSE.pth ├── pipeline_flux_4k_wavelet.py ├── pipeline_sd3_4k_wavelet.py ├── requirements.txt ├── sc-vae ├── README.md ├── autoencoder_kl.py ├── config.yaml ├── data │ └── sa-1b │ │ └── example_metadata_vae_sa1b.jsonl ├── discriminator_vae.py ├── train_vae.py ├── train_vae.sh └── vae.py ├── test_flux.py ├── test_sd3.py ├── train_flux.sh ├── train_flux_4k_wavelet.py ├── train_sd3.sh ├── train_sd3_4k_wavelet.py └── vae.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/README.md -------------------------------------------------------------------------------- /asset/dataset.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/asset/dataset.jpg -------------------------------------------------------------------------------- /asset/demo.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/asset/demo.jpg -------------------------------------------------------------------------------- /asset/sc-vae.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/asset/sc-vae.png -------------------------------------------------------------------------------- /autoencoder_kl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/autoencoder_kl.py -------------------------------------------------------------------------------- /ds_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/ds_config.yaml -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/eval.py -------------------------------------------------------------------------------- /generate_and_eval_fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/generate_and_eval_fid.py -------------------------------------------------------------------------------- /improved-aesthetic-predictor/sac+logos+ava1-l14-linearMSE.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/improved-aesthetic-predictor/sac+logos+ava1-l14-linearMSE.pth -------------------------------------------------------------------------------- /pipeline_flux_4k_wavelet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/pipeline_flux_4k_wavelet.py -------------------------------------------------------------------------------- /pipeline_sd3_4k_wavelet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/pipeline_sd3_4k_wavelet.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/requirements.txt -------------------------------------------------------------------------------- /sc-vae/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/sc-vae/README.md -------------------------------------------------------------------------------- /sc-vae/autoencoder_kl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/sc-vae/autoencoder_kl.py -------------------------------------------------------------------------------- /sc-vae/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/sc-vae/config.yaml -------------------------------------------------------------------------------- /sc-vae/data/sa-1b/example_metadata_vae_sa1b.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/sc-vae/data/sa-1b/example_metadata_vae_sa1b.jsonl -------------------------------------------------------------------------------- /sc-vae/discriminator_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/sc-vae/discriminator_vae.py -------------------------------------------------------------------------------- /sc-vae/train_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/sc-vae/train_vae.py -------------------------------------------------------------------------------- /sc-vae/train_vae.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/sc-vae/train_vae.sh -------------------------------------------------------------------------------- /sc-vae/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/sc-vae/vae.py -------------------------------------------------------------------------------- /test_flux.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/test_flux.py -------------------------------------------------------------------------------- /test_sd3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/test_sd3.py -------------------------------------------------------------------------------- /train_flux.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/train_flux.sh -------------------------------------------------------------------------------- /train_flux_4k_wavelet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/train_flux_4k_wavelet.py -------------------------------------------------------------------------------- /train_sd3.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/train_sd3.sh -------------------------------------------------------------------------------- /train_sd3_4k_wavelet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/train_sd3_4k_wavelet.py -------------------------------------------------------------------------------- /vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhang0jhon/diffusion-4k/HEAD/vae.py --------------------------------------------------------------------------------