├── .gitignore ├── README.md ├── batch_files ├── templates │ ├── aws_slurm.sh │ └── slurm.sh └── tests │ └── wds_seed_test.sbatch ├── configs └── templates │ ├── imagenet.yaml │ ├── sd_v2_base_stage1.yaml │ └── sd_v2_base_stage2.yaml ├── data ├── __init__.py ├── base.py ├── metrics.py ├── policies.py └── prompts │ ├── imagenet_captions.csv │ ├── uid_caption.csv │ ├── uid_caption_reformat.csv │ └── uid_caption_short.csv ├── experimental └── train_ae.py ├── images └── imagenet-examples.png ├── requirements.txt ├── scripts ├── download_and_tar_wds.py ├── download_diffusers.py ├── extract_wds_metadata.py ├── generate_examples.py ├── metadata │ ├── imagenet_classnames.json │ └── imagenet_idx_to_prompt.json └── test_on_multinode.py ├── train.py └── utils ├── __init__.py ├── data_utils.py ├── distributed.py ├── getters.py ├── logging.py ├── net_utils.py ├── optimizers.py └── schedulers.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/README.md -------------------------------------------------------------------------------- /batch_files/templates/aws_slurm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/batch_files/templates/aws_slurm.sh -------------------------------------------------------------------------------- /batch_files/templates/slurm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/batch_files/templates/slurm.sh -------------------------------------------------------------------------------- /batch_files/tests/wds_seed_test.sbatch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/batch_files/tests/wds_seed_test.sbatch -------------------------------------------------------------------------------- /configs/templates/imagenet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/configs/templates/imagenet.yaml -------------------------------------------------------------------------------- /configs/templates/sd_v2_base_stage1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/configs/templates/sd_v2_base_stage1.yaml -------------------------------------------------------------------------------- /configs/templates/sd_v2_base_stage2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/configs/templates/sd_v2_base_stage2.yaml -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/data/base.py -------------------------------------------------------------------------------- /data/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/data/metrics.py -------------------------------------------------------------------------------- /data/policies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/data/policies.py -------------------------------------------------------------------------------- /data/prompts/imagenet_captions.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/data/prompts/imagenet_captions.csv -------------------------------------------------------------------------------- /data/prompts/uid_caption.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/data/prompts/uid_caption.csv -------------------------------------------------------------------------------- /data/prompts/uid_caption_reformat.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/data/prompts/uid_caption_reformat.csv -------------------------------------------------------------------------------- /data/prompts/uid_caption_short.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/data/prompts/uid_caption_short.csv -------------------------------------------------------------------------------- /experimental/train_ae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/experimental/train_ae.py -------------------------------------------------------------------------------- /images/imagenet-examples.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/images/imagenet-examples.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/download_and_tar_wds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/scripts/download_and_tar_wds.py -------------------------------------------------------------------------------- /scripts/download_diffusers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/scripts/download_diffusers.py -------------------------------------------------------------------------------- /scripts/extract_wds_metadata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/scripts/extract_wds_metadata.py -------------------------------------------------------------------------------- /scripts/generate_examples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/scripts/generate_examples.py -------------------------------------------------------------------------------- /scripts/metadata/imagenet_classnames.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/scripts/metadata/imagenet_classnames.json -------------------------------------------------------------------------------- /scripts/metadata/imagenet_idx_to_prompt.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/scripts/metadata/imagenet_idx_to_prompt.json -------------------------------------------------------------------------------- /scripts/test_on_multinode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/scripts/test_on_multinode.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/utils/data_utils.py -------------------------------------------------------------------------------- /utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/utils/distributed.py -------------------------------------------------------------------------------- /utils/getters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/utils/getters.py -------------------------------------------------------------------------------- /utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/utils/logging.py -------------------------------------------------------------------------------- /utils/net_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/utils/net_utils.py -------------------------------------------------------------------------------- /utils/optimizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/utils/optimizers.py -------------------------------------------------------------------------------- /utils/schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlfoundations/open-diffusion/HEAD/utils/schedulers.py --------------------------------------------------------------------------------