LICENSE
README.md
setup.py
cad/__init__.py
cad/pipe.py
cad/sample_image.py
cad/test.py
cad/train.py
cad/validate.py
cad/__pycache__/__init__.cpython-310.pyc
cad/__pycache__/__init__.cpython-311.pyc
cad/__pycache__/pipe.cpython-310.pyc
cad/__pycache__/pipe.cpython-311.pyc
cad/callbacks/__init__.py
cad/callbacks/checkpoint_and_validate.py
cad/callbacks/data.py
cad/callbacks/ema.py
cad/callbacks/fix_nans.py
cad/callbacks/log_images.py
cad/callbacks/__pycache__/__init__.cpython-310.pyc
cad/callbacks/__pycache__/checkpoint_and_validate.cpython-310.pyc
cad/callbacks/__pycache__/data.cpython-310.pyc
cad/callbacks/__pycache__/ema.cpython-310.pyc
cad/callbacks/__pycache__/fid.cpython-310.pyc
cad/callbacks/__pycache__/fix_nans.cpython-310.pyc
cad/callbacks/__pycache__/log_images.cpython-310.pyc
cad/configs/config.yaml
cad/configs/computer/a100.yaml
cad/configs/computer/cluster-node-a100.yaml
cad/configs/computer/cluster-node-a100_fsdp.yaml
cad/configs/computer/cluster-node-v100.yaml
cad/configs/computer/cpu.yaml
cad/configs/computer/v100.yaml
cad/configs/data/default.yaml
cad/configs/data/webdataset.yaml
cad/configs/data/collate_fn/default.yaml
cad/configs/data/collate_fn/from_dict_pad.yaml
cad/configs/data/collate_fn/from_tuple.yaml
cad/configs/data/dataset/cc12m.yaml
cad/configs/data/dataset/cc12m_la6.yaml
cad/configs/data/dataset/cc3m.yaml
cad/configs/data/dataset/cifar10.yaml
cad/configs/data/dataset/imagenet.yaml
cad/configs/data/dataset/imagenet_ldm.yaml
cad/configs/data/dataset/imagenet_ldm_text.yaml
cad/configs/data/train_aug/center_crop.yaml
cad/configs/data/train_aug/none.yaml
cad/configs/data/val_aug/center_crop.yaml
cad/configs/data/val_aug/none.yaml
cad/configs/model/cad_text_cond.yaml
cad/configs/model/class_cond.yaml
cad/configs/model/text_cond.yaml
cad/configs/model/cond_preprocessing/label.yaml
cad/configs/model/cond_preprocessing/precomputed_text_emb.yaml
cad/configs/model/cond_preprocessing/text.yaml
cad/configs/model/data_preprocessing/pixels.yaml
cad/configs/model/data_preprocessing/precomputed_sd_latents.yaml
cad/configs/model/data_preprocessing/precomputed_sd_latents_resample.yaml
cad/configs/model/data_preprocessing/sd_latents.yaml
cad/configs/model/inference_noise_scheduler/cosine.yaml
cad/configs/model/inference_noise_scheduler/cosine_simple.yaml
cad/configs/model/inference_noise_scheduler/linear.yaml
cad/configs/model/inference_noise_scheduler/sigmoid.yaml
cad/configs/model/loss/ddpm.yaml
cad/configs/model/loss/edm.yaml
cad/configs/model/loss/ve.yaml
cad/configs/model/loss/vp.yaml
cad/configs/model/lr_scheduler/warmup.yaml
cad/configs/model/lr_scheduler/warmup_cosine_decay.yaml
cad/configs/model/lr_scheduler/warmup_sqrt_decay.yaml
cad/configs/model/network/cad_rin_text.yaml
cad/configs/model/network/ddpm++.yaml
cad/configs/model/network/ncsn++.yaml
cad/configs/model/network/rin.yaml
cad/configs/model/network/rin_text.yaml
cad/configs/model/optimizer/adam.yaml
cad/configs/model/optimizer/adamw.yaml
cad/configs/model/optimizer/lamb.yaml
cad/configs/model/optimizer/sgd.yaml
cad/configs/model/postprocessing/pixels.yaml
cad/configs/model/postprocessing/sd_1_5_consistency.yaml
cad/configs/model/postprocessing/sd_1_5_vae.yaml
cad/configs/model/postprocessing/sd_xl_vae.yaml
cad/configs/model/precomputed_text_embeddings/no.yaml
cad/configs/model/precomputed_text_embeddings/yes.yaml
cad/configs/model/preconditioning/ddpm.yaml
cad/configs/model/preconditioning/edm.yaml
cad/configs/model/preconditioning/ve.yaml
cad/configs/model/preconditioning/vp.yaml
cad/configs/model/test_sampler/ddim.yaml
cad/configs/model/test_sampler/ddpm.yaml
cad/configs/model/test_sampler/dpm.yaml
cad/configs/model/test_sampler/edm.yaml
cad/configs/model/train_noise_scheduler/cosine.yaml
cad/configs/model/train_noise_scheduler/cosine_simple.yaml
cad/configs/model/train_noise_scheduler/linear.yaml
cad/configs/model/train_noise_scheduler/sigmoid.yaml
cad/configs/model/uncond_conditioning/empty_string.yaml
cad/configs/model/uncond_conditioning/t5_uncond.yaml
cad/configs/model/uncond_conditioning/zero.yaml
cad/configs/model/use_ldm/no.yaml
cad/configs/model/use_ldm/precomputed.yaml
cad/configs/model/use_ldm/precomputed_resample.yaml
cad/configs/model/use_ldm/yes.yaml
cad/configs/model/val_sampler/ddim.yaml
cad/configs/model/val_sampler/ddpm.yaml
cad/configs/model/val_sampler/dpm.yaml
cad/configs/model/val_sampler/edm.yaml
cad/configs/overrides/cad_256.yaml
cad/configs/overrides/cad_512.yaml
cad/configs/overrides/cc12m_256_rin.yaml
cad/configs/overrides/cc12m_256_rin_small_ldm.yaml
cad/configs/overrides/cc12m_256_rin_small_ldm_filtered.yaml
cad/configs/overrides/cc12m_256_rin_small_ldm_weighted.yaml
cad/configs/overrides/cc12m_256_rin_tiny_ldm.yaml
cad/configs/overrides/cc12m_256_rin_tiny_ldm_cad.yaml
cad/configs/overrides/cc12m_256_rin_tiny_ldm_filtered.yaml
cad/configs/overrides/cc12m_256_rin_tiny_ldm_weighted.yaml
cad/configs/overrides/cc12m_512_rin_ldm.yaml
cad/configs/overrides/cc12m_512_rin_ldm_cad.yaml
cad/configs/overrides/cc12m_512_rin_small_ldm_cad.yaml
cad/configs/overrides/cc12m_64_rin.yaml
cad/configs/overrides/cc12m_64_rin_small.yaml
cad/configs/overrides/cc3m_64_rin.yaml
cad/configs/overrides/cifar10_rin.yaml
cad/configs/overrides/imagenet_256_rin_base_text_ldm.yaml
cad/configs/overrides/imagenet_256_rin_ldm.yaml
cad/configs/overrides/imagenet_256_rin_ldm_large.yaml
cad/configs/overrides/imagenet_64_rin.yaml
cad/configs/stage/debug.yaml
cad/configs/stage/profile.yaml
cad/configs/use_condition/cad_guidance.yaml
cad/configs/use_condition/classifier_free_guidance.yaml
cad/configs/use_condition/conditional.yaml
cad/configs/use_condition/unconditional.yaml
cad/data/__init__.py
cad/data/datamodule.py
cad/data/dataset.py
cad/data/text_dataset.py
cad/data/__pycache__/__init__.cpython-310.pyc
cad/data/__pycache__/datamodule.cpython-310.pyc
cad/data/__pycache__/dataset.cpython-310.pyc
cad/data/__pycache__/resampled_dataset.cpython-310.pyc
cad/data/__pycache__/text_dataset.cpython-310.pyc
cad/data/__pycache__/utils.cpython-310.pyc
cad/data/processing_scripts/__init__.py
cad/data/processing_scripts/coco_to_wds.py
cad/data/processing_scripts/compute_aesthetics.py
cad/data/processing_scripts/compute_clip_scores.py
cad/data/processing_scripts/compute_t5_embeddings.py
cad/data/processing_scripts/compute_vae.py
cad/data/processing_scripts/create_prompt_bed.py
cad/data/processing_scripts/imagenet_simple_caption.py
cad/data/processing_scripts/imagenet_to_ldm_webdataset.py
cad/data/processing_scripts/launch_aesthetics.py
cad/data/processing_scripts/launch_clip_scoring.py
cad/data/processing_scripts/launch_preprocess_data.py
cad/data/processing_scripts/launch_t5_embs_storage.py
cad/data/processing_scripts/make_ldm_imagenet.py
cad/data/processing_scripts/preprocess_data.py
cad/data/processing_scripts/preprocess_synthetic.py
cad/data/processing_scripts/webdataset_to_dataset.py
cad/data/processing_scripts/webdataset_to_dataset_class_cond.py
cad/data/synthetic_captions/__init__.py
cad/data/synthetic_captions/captioner.py
cad/data/synthetic_captions/captioners/id_to_caption.py
cad/jean_zay/__init__.py
cad/jean_zay/launch.py
cad/jean_zay/launch_cfg_models.py
cad/jean_zay/launch_cfg_models_imagenet.py
cad/jean_zay/launch_confidence_eval.py
cad/jean_zay/launch_imagenet_ldm.py
cad/jean_zay/launch_single_run.py
cad/jean_zay/launch_single_run_test.py
cad/jean_zay/launch_text2im_256.py
cad/jean_zay/launch_text2img.py
cad/jean_zay/launch_text2img_ldm.py
cad/jean_zay/launch_text2img_ldm_bin_ablation.py
cad/jean_zay/launch_text2img_ldm_small.py
cad/jean_zay/launch_text2img_ldm_small_512_finetune.py
cad/jean_zay/launch_text2img_ldm_tiny.py
cad/jean_zay/launch_text2img_small.py
cad/jean_zay/load_all_models_cache.py
cad/jean_zay/__pycache__/__init__.cpython-310.pyc
cad/jean_zay/__pycache__/launch.cpython-310.pyc
cad/jean_zay/__pycache__/launch_single_run_test.cpython-310-pytest-8.3.3.pyc
cad/metrics/__init__.py
cad/metrics/classifier_acc.py
cad/metrics/inception_metrics.py
cad/metrics/sample_and_eval.py
cad/metrics/__pycache__/__init__.cpython-310.pyc
cad/metrics/__pycache__/classifier_acc.cpython-310.pyc
cad/metrics/__pycache__/inception_metrics.cpython-310.pyc
cad/metrics/__pycache__/sample_and_eval.cpython-310.pyc
cad/models/__init__.py
cad/models/diffusion.py
cad/models/losses.py
cad/models/positional_embeddings.py
cad/models/postprocessing.py
cad/models/preconditioning.py
cad/models/preprocessing.py
cad/models/pretrained_models.py
cad/models/schedulers.py
cad/models/__pycache__/__init__.cpython-310.pyc
cad/models/__pycache__/__init__.cpython-311.pyc
cad/models/__pycache__/__init__.cpython-38.pyc
cad/models/__pycache__/__init__.cpython-39.pyc
cad/models/__pycache__/diffusion.cpython-310.pyc
cad/models/__pycache__/diffusion.cpython-38.pyc
cad/models/__pycache__/diffusion.cpython-39.pyc
cad/models/__pycache__/losses.cpython-310.pyc
cad/models/__pycache__/noise_encoding.cpython-310.pyc
cad/models/__pycache__/positional_embeddings.cpython-310.pyc
cad/models/__pycache__/positional_embeddings.cpython-311.pyc
cad/models/__pycache__/postprocessing.cpython-310.pyc
cad/models/__pycache__/postprocessing.cpython-311.pyc
cad/models/__pycache__/preconditioning.cpython-310.pyc
cad/models/__pycache__/preconditioning.cpython-311.pyc
cad/models/__pycache__/preprocessing.cpython-310.pyc
cad/models/__pycache__/preprocessing.cpython-311.pyc
cad/models/__pycache__/pretrained_models.cpython-310.pyc
cad/models/__pycache__/pretrained_models.cpython-311.pyc
cad/models/__pycache__/schedulers.cpython-310.pyc
cad/models/__pycache__/schedulers.cpython-311.pyc
cad/models/networks/__init__.py
cad/models/networks/ca_unet.py
cad/models/networks/rin.py
cad/models/networks/transformers.py
cad/models/networks/unet.py
cad/models/networks/utils.py
cad/models/networks/__pycache__/__init__.cpython-310.pyc
cad/models/networks/__pycache__/__init__.cpython-311.pyc
cad/models/networks/__pycache__/packed_rin.cpython-310.pyc
cad/models/networks/__pycache__/packed_transformers.cpython-310.pyc
cad/models/networks/__pycache__/rin.cpython-310.pyc
cad/models/networks/__pycache__/rin.cpython-311.pyc
cad/models/networks/__pycache__/transformers.cpython-310.pyc
cad/models/networks/__pycache__/transformers.cpython-311.pyc
cad/models/networks/__pycache__/unet.cpython-310.pyc
cad/models/networks/__pycache__/utils.cpython-310.pyc
cad/models/samplers/__init__.py
cad/models/samplers/base_sampler.py
cad/models/samplers/ddim.py
cad/models/samplers/ddpm.py
cad/models/samplers/dpm.py
cad/models/samplers/dpm_euler.py
cad/models/samplers/dpm_plusplus_2M.py
cad/models/samplers/dpm_plusplus_2S.py
cad/models/samplers/edm.py
cad/models/samplers/__pycache__/__init__.cpython-310.pyc
cad/models/samplers/__pycache__/__init__.cpython-311.pyc
cad/models/samplers/__pycache__/base_sampler.cpython-310.pyc
cad/models/samplers/__pycache__/base_sampler.cpython-311.pyc
cad/models/samplers/__pycache__/ddim.cpython-310.pyc
cad/models/samplers/__pycache__/ddim.cpython-311.pyc
cad/models/samplers/__pycache__/ddim_legacy.cpython-310.pyc
cad/models/samplers/__pycache__/ddim_new.cpython-310.pyc
cad/models/samplers/__pycache__/ddim_sampler.cpython-310.pyc
cad/models/samplers/__pycache__/ddpm.cpython-310.pyc
cad/models/samplers/__pycache__/ddpm.cpython-311.pyc
cad/models/samplers/__pycache__/ddpm_legacy.cpython-310.pyc
cad/models/samplers/__pycache__/ddpm_new.cpython-310.pyc
cad/models/samplers/__pycache__/ddpm_sampler.cpython-310.pyc
cad/models/samplers/__pycache__/dpm.cpython-310.pyc
cad/models/samplers/__pycache__/dpm.cpython-311.pyc
cad/models/samplers/__pycache__/dpm_euler.cpython-310.pyc
cad/models/samplers/__pycache__/dpm_euler.cpython-311.pyc
cad/models/samplers/__pycache__/dpm_legacy.cpython-310.pyc
cad/models/samplers/__pycache__/dpm_new.cpython-310.pyc
cad/models/samplers/__pycache__/dpm_plusplus_2M.cpython-310.pyc
cad/models/samplers/__pycache__/dpm_plusplus_2M.cpython-311.pyc
cad/models/samplers/__pycache__/dpm_plusplus_2S.cpython-310.pyc
cad/models/samplers/__pycache__/dpm_plusplus_2S.cpython-311.pyc
cad/models/samplers/__pycache__/dpm_sampler.cpython-310.pyc
cad/models/samplers/__pycache__/edm.cpython-310.pyc
cad/utils/__init__.py
cad/utils/aesthetic_scorer.py
cad/utils/flan_t5_xl_random.npy
cad/utils/flan_t5_xl_uncond.npy
cad/utils/graft_model.py
cad/utils/hotfix_checkpoints.py
cad/utils/image_processing.py
cad/utils/lr_scheduler.py
cad/utils/misc.py
cad/utils/one_hot_transform.py
cad/utils/optimizers.py
cad/utils/__pycache__/__init__.cpython-310.pyc
cad/utils/__pycache__/__init__.cpython-311.pyc
cad/utils/__pycache__/__init__.cpython-38.pyc
cad/utils/__pycache__/aesthetic_scorer.cpython-310.pyc
cad/utils/__pycache__/entropy.cpython-310.pyc
cad/utils/__pycache__/hpsv2.cpython-310.pyc
cad/utils/__pycache__/image_processing.cpython-310.pyc
cad/utils/__pycache__/image_processing.cpython-311.pyc
cad/utils/__pycache__/image_processing.cpython-38.pyc
cad/utils/__pycache__/lr_scheduler.cpython-310.pyc
cad/utils/__pycache__/misc.cpython-310.pyc
cad/utils/__pycache__/one_hot_transform.cpython-310.pyc
cad/utils/__pycache__/optimizers.cpython-310.pyc
cad_diffusion.egg-info/PKG-INFO
cad_diffusion.egg-info/SOURCES.txt
cad_diffusion.egg-info/dependency_links.txt
cad_diffusion.egg-info/requires.txt
cad_diffusion.egg-info/top_level.txt