├── DiT ├── coordcheck.py ├── create_npz.py ├── diffusion │ ├── __init__.py │ ├── diffusion_utils.py │ ├── gaussian_diffusion.py │ ├── respace.py │ └── timestep_sampler.py ├── download.py ├── environment.yml ├── evaluator.py ├── models.py ├── models_mup.py ├── sample_ddp.py ├── scripts │ ├── coordcheck.sh │ ├── pretraining_mup.sh │ ├── pretraining_original.sh │ ├── sample.sh │ └── search_hp.sh ├── train_mup.py ├── utils │ ├── __init__.py │ ├── coord_check.py │ ├── logger.py │ └── seed.py └── width288_d28.bsh ├── LICENSE ├── PixArt-alpha ├── L28_width288.bsh ├── configs │ ├── PixArt_xl2_sam.py │ └── pixart_config │ │ ├── PixArt_mup_img256_SAM_coord.py │ │ ├── PixArt_mup_xl2_img256_SAM_proxy.py │ │ ├── PixArt_mup_xl2_img256_SAM_target.py │ │ └── PixArt_xl2_img256_SAM.py ├── coord_check.sh ├── diffusion │ ├── __init__.py │ ├── data │ │ ├── __init__.py │ │ ├── builder.py │ │ ├── datasets │ │ │ ├── SA.py │ │ │ ├── __init__.py │ │ │ └── utils.py │ │ └── transforms.py │ ├── dpm_solver.py │ ├── iddpm.py │ ├── lcm_scheduler.py │ ├── model │ │ ├── __init__.py │ │ ├── builder.py │ │ ├── diffusion_utils.py │ │ ├── dpm_solver.py │ │ ├── edm_sample.py │ │ ├── gaussian_diffusion.py │ │ ├── hed.py │ │ ├── nets │ │ │ ├── PixArt.py │ │ │ ├── PixArt_blocks.py │ │ │ └── __init__.py │ │ ├── respace.py │ │ ├── sa_solver.py │ │ ├── t5.py │ │ ├── timestep_sampler.py │ │ └── utils.py │ ├── sa_sampler.py │ ├── sa_solver_diffusers.py │ └── utils │ │ ├── __init__.py │ │ ├── checkpoint.py │ │ ├── data_sampler.py │ │ ├── dist_utils.py │ │ ├── logger.py │ │ ├── lr_scheduler.py │ │ ├── misc.py │ │ └── optimizer.py ├── inference.sh ├── inference_geneval.sh ├── pretraining_mup.sh ├── pretraining_original.sh ├── requirements.txt ├── scripts │ ├── coordcheck.py │ ├── inference.py │ └── inference_geneval.py ├── search_hp.sh ├── tools │ ├── clip_score.py │ ├── coord_check.py │ ├── data_util.py │ ├── download.py │ ├── evaluate_geneval.py │ ├── extract_features.py │ ├── fid.py │ ├── geneval-main │ │ ├── LICENSE │ │ ├── README.md │ │ ├── annotations │ │ │ ├── annotations_clip.csv │ │ │ ├── annotations_if-xl.csv │ │ │ ├── annotations_sdv2.csv │ │ │ └── mturk_hit_template.html │ │ ├── environment.yml │ │ ├── evaluation │ │ │ └── download_models.sh │ │ ├── generation │ │ │ └── diffusers_generate copy.py │ │ ├── images │ │ │ └── geneval_figure_1.png │ │ └── prompts │ │ │ ├── create_prompts.py │ │ │ ├── evaluation_metadata.jsonl │ │ │ ├── generation_prompts.txt │ │ │ └── object_names.txt │ ├── object_names.txt │ └── summary_scores.py └── train_scripts │ └── train.py ├── README.md └── assets ├── HP_transfer.jpg ├── dit_coord.jpg ├── fid_pretraining_dit.png ├── pixart_coord.jpg └── vis.jpg /DiT/coordcheck.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/DiT/coordcheck.py -------------------------------------------------------------------------------- /DiT/create_npz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/DiT/create_npz.py -------------------------------------------------------------------------------- /DiT/diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/DiT/diffusion/__init__.py -------------------------------------------------------------------------------- /DiT/diffusion/diffusion_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/DiT/diffusion/diffusion_utils.py -------------------------------------------------------------------------------- /DiT/diffusion/gaussian_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/DiT/diffusion/gaussian_diffusion.py -------------------------------------------------------------------------------- /DiT/diffusion/respace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/DiT/diffusion/respace.py -------------------------------------------------------------------------------- /DiT/diffusion/timestep_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/DiT/diffusion/timestep_sampler.py -------------------------------------------------------------------------------- /DiT/download.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/DiT/download.py -------------------------------------------------------------------------------- /DiT/environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/DiT/environment.yml -------------------------------------------------------------------------------- /DiT/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/DiT/evaluator.py -------------------------------------------------------------------------------- /DiT/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/DiT/models.py -------------------------------------------------------------------------------- /DiT/models_mup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/DiT/models_mup.py -------------------------------------------------------------------------------- /DiT/sample_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/DiT/sample_ddp.py -------------------------------------------------------------------------------- /DiT/scripts/coordcheck.sh: -------------------------------------------------------------------------------- 1 | python coordcheck.py --load_base_shapes width288_d28.bsh -------------------------------------------------------------------------------- /DiT/scripts/pretraining_mup.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/DiT/scripts/pretraining_mup.sh -------------------------------------------------------------------------------- /DiT/scripts/pretraining_original.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/DiT/scripts/pretraining_original.sh -------------------------------------------------------------------------------- /DiT/scripts/sample.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/DiT/scripts/sample.sh -------------------------------------------------------------------------------- /DiT/scripts/search_hp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/DiT/scripts/search_hp.sh -------------------------------------------------------------------------------- /DiT/train_mup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/DiT/train_mup.py -------------------------------------------------------------------------------- /DiT/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/DiT/utils/__init__.py -------------------------------------------------------------------------------- /DiT/utils/coord_check.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/DiT/utils/coord_check.py -------------------------------------------------------------------------------- /DiT/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/DiT/utils/logger.py -------------------------------------------------------------------------------- /DiT/utils/seed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/DiT/utils/seed.py -------------------------------------------------------------------------------- /DiT/width288_d28.bsh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/DiT/width288_d28.bsh -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/LICENSE -------------------------------------------------------------------------------- /PixArt-alpha/L28_width288.bsh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/L28_width288.bsh -------------------------------------------------------------------------------- /PixArt-alpha/configs/PixArt_xl2_sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/configs/PixArt_xl2_sam.py -------------------------------------------------------------------------------- /PixArt-alpha/configs/pixart_config/PixArt_mup_img256_SAM_coord.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/configs/pixart_config/PixArt_mup_img256_SAM_coord.py -------------------------------------------------------------------------------- /PixArt-alpha/configs/pixart_config/PixArt_mup_xl2_img256_SAM_proxy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/configs/pixart_config/PixArt_mup_xl2_img256_SAM_proxy.py -------------------------------------------------------------------------------- /PixArt-alpha/configs/pixart_config/PixArt_mup_xl2_img256_SAM_target.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/configs/pixart_config/PixArt_mup_xl2_img256_SAM_target.py -------------------------------------------------------------------------------- /PixArt-alpha/configs/pixart_config/PixArt_xl2_img256_SAM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/configs/pixart_config/PixArt_xl2_img256_SAM.py -------------------------------------------------------------------------------- /PixArt-alpha/coord_check.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/coord_check.sh -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/__init__.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/data/__init__.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/data/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/data/builder.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/data/datasets/SA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/data/datasets/SA.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/data/datasets/__init__.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/data/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/data/datasets/utils.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/data/transforms.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/dpm_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/dpm_solver.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/iddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/iddpm.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/lcm_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/lcm_scheduler.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/model/__init__.py: -------------------------------------------------------------------------------- 1 | from .nets import * 2 | -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/model/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/model/builder.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/model/diffusion_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/model/diffusion_utils.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/model/dpm_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/model/dpm_solver.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/model/edm_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/model/edm_sample.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/model/gaussian_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/model/gaussian_diffusion.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/model/hed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/model/hed.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/model/nets/PixArt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/model/nets/PixArt.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/model/nets/PixArt_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/model/nets/PixArt_blocks.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/model/nets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/model/nets/__init__.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/model/respace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/model/respace.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/model/sa_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/model/sa_solver.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/model/t5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/model/t5.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/model/timestep_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/model/timestep_sampler.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/model/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/model/utils.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/sa_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/sa_sampler.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/sa_solver_diffusers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/sa_solver_diffusers.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/utils/checkpoint.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/utils/data_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/utils/data_sampler.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/utils/dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/utils/dist_utils.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/utils/logger.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/utils/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/utils/lr_scheduler.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/utils/misc.py -------------------------------------------------------------------------------- /PixArt-alpha/diffusion/utils/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/diffusion/utils/optimizer.py -------------------------------------------------------------------------------- /PixArt-alpha/inference.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/inference.sh -------------------------------------------------------------------------------- /PixArt-alpha/inference_geneval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/inference_geneval.sh -------------------------------------------------------------------------------- /PixArt-alpha/pretraining_mup.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/pretraining_mup.sh -------------------------------------------------------------------------------- /PixArt-alpha/pretraining_original.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/pretraining_original.sh -------------------------------------------------------------------------------- /PixArt-alpha/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/requirements.txt -------------------------------------------------------------------------------- /PixArt-alpha/scripts/coordcheck.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/scripts/coordcheck.py -------------------------------------------------------------------------------- /PixArt-alpha/scripts/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/scripts/inference.py -------------------------------------------------------------------------------- /PixArt-alpha/scripts/inference_geneval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/scripts/inference_geneval.py -------------------------------------------------------------------------------- /PixArt-alpha/search_hp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/search_hp.sh -------------------------------------------------------------------------------- /PixArt-alpha/tools/clip_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/tools/clip_score.py -------------------------------------------------------------------------------- /PixArt-alpha/tools/coord_check.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/tools/coord_check.py -------------------------------------------------------------------------------- /PixArt-alpha/tools/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/tools/data_util.py -------------------------------------------------------------------------------- /PixArt-alpha/tools/download.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/tools/download.py -------------------------------------------------------------------------------- /PixArt-alpha/tools/evaluate_geneval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/tools/evaluate_geneval.py -------------------------------------------------------------------------------- /PixArt-alpha/tools/extract_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/tools/extract_features.py -------------------------------------------------------------------------------- /PixArt-alpha/tools/fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/tools/fid.py -------------------------------------------------------------------------------- /PixArt-alpha/tools/geneval-main/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/tools/geneval-main/LICENSE -------------------------------------------------------------------------------- /PixArt-alpha/tools/geneval-main/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/tools/geneval-main/README.md -------------------------------------------------------------------------------- /PixArt-alpha/tools/geneval-main/annotations/annotations_clip.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/tools/geneval-main/annotations/annotations_clip.csv -------------------------------------------------------------------------------- /PixArt-alpha/tools/geneval-main/annotations/annotations_if-xl.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/tools/geneval-main/annotations/annotations_if-xl.csv -------------------------------------------------------------------------------- /PixArt-alpha/tools/geneval-main/annotations/annotations_sdv2.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/tools/geneval-main/annotations/annotations_sdv2.csv -------------------------------------------------------------------------------- /PixArt-alpha/tools/geneval-main/annotations/mturk_hit_template.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/tools/geneval-main/annotations/mturk_hit_template.html -------------------------------------------------------------------------------- /PixArt-alpha/tools/geneval-main/environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/tools/geneval-main/environment.yml -------------------------------------------------------------------------------- /PixArt-alpha/tools/geneval-main/evaluation/download_models.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/tools/geneval-main/evaluation/download_models.sh -------------------------------------------------------------------------------- /PixArt-alpha/tools/geneval-main/generation/diffusers_generate copy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/tools/geneval-main/generation/diffusers_generate copy.py -------------------------------------------------------------------------------- /PixArt-alpha/tools/geneval-main/images/geneval_figure_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/tools/geneval-main/images/geneval_figure_1.png -------------------------------------------------------------------------------- /PixArt-alpha/tools/geneval-main/prompts/create_prompts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/tools/geneval-main/prompts/create_prompts.py -------------------------------------------------------------------------------- /PixArt-alpha/tools/geneval-main/prompts/evaluation_metadata.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/tools/geneval-main/prompts/evaluation_metadata.jsonl -------------------------------------------------------------------------------- /PixArt-alpha/tools/geneval-main/prompts/generation_prompts.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/tools/geneval-main/prompts/generation_prompts.txt -------------------------------------------------------------------------------- /PixArt-alpha/tools/geneval-main/prompts/object_names.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/tools/geneval-main/prompts/object_names.txt -------------------------------------------------------------------------------- /PixArt-alpha/tools/object_names.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/tools/object_names.txt -------------------------------------------------------------------------------- /PixArt-alpha/tools/summary_scores.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/tools/summary_scores.py -------------------------------------------------------------------------------- /PixArt-alpha/train_scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/PixArt-alpha/train_scripts/train.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/README.md -------------------------------------------------------------------------------- /assets/HP_transfer.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/assets/HP_transfer.jpg -------------------------------------------------------------------------------- /assets/dit_coord.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/assets/dit_coord.jpg -------------------------------------------------------------------------------- /assets/fid_pretraining_dit.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/assets/fid_pretraining_dit.png -------------------------------------------------------------------------------- /assets/pixart_coord.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/assets/pixart_coord.jpg -------------------------------------------------------------------------------- /assets/vis.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ML-GSAI/Scaling-Diffusion-Transformers-muP/HEAD/assets/vis.jpg --------------------------------------------------------------------------------