├── .gitignore ├── LICENSE ├── README.md ├── applications ├── dc_ae │ ├── dc_ae_generate_latent.py │ ├── demo_dc_ae_diffusion_model.py │ ├── demo_dc_ae_model.py │ ├── demo_dc_ae_model_diffusers.py │ ├── eval_dc_ae_diffusion_model.py │ ├── eval_dc_ae_model.py │ ├── generate_reference.py │ └── train_dc_ae_diffusion_model.py └── dc_t2i │ ├── get_null_embedding.py │ └── manual_prompt.py ├── assets ├── dc-gen-figures │ ├── analysis.jpg │ ├── appendix_1K.png │ ├── comp_qualitative.jpg │ ├── comp_speed.jpg │ ├── pipeline.jpg │ └── teaser.jpg ├── docs │ └── dc_ae_sana_1.1.md ├── train_diffusion_with_flux_vae_f8.md └── train_diffusion_with_sd_vae_f8.md ├── dc_gen ├── __init__.py ├── ae_model_zoo.py ├── aecore │ ├── __init__.py │ ├── data_provider │ │ ├── __init__.py │ │ ├── base.py │ │ ├── collection.py │ │ ├── examine.py │ │ ├── imagenet.py │ │ ├── mixture.py │ │ ├── mjhq.py │ │ └── vlm_caption.py │ ├── efficiency │ │ └── test.py │ ├── models │ │ ├── __init__.py │ │ ├── base.py │ │ ├── dc_ae.py │ │ ├── dc_ae_diffusers.py │ │ ├── hf_autoencoder_kl.py │ │ └── sd_vae.py │ └── trainer.py ├── apps │ ├── __init__.py │ ├── data_provider │ │ ├── __init__.py │ │ ├── dc_base.py │ │ ├── dc_mixture.py │ │ ├── generate_archives.py │ │ ├── generate_latent_archives.py │ │ ├── sampler.py │ │ ├── utils.py │ │ ├── web_dataset │ │ │ ├── __init__.py │ │ │ ├── generate_meta.py │ │ │ ├── wids.py │ │ │ ├── wids_dl.py │ │ │ ├── wids_lru.py │ │ │ ├── wids_mmtar.py │ │ │ ├── wids_specs.py │ │ │ └── wids_tar.py │ │ └── zip_dataset │ │ │ ├── __init__.py │ │ │ ├── generate_meta.py │ │ │ ├── single_zip_dataset.py │ │ │ └── zip_dataset.py │ ├── metrics │ │ ├── __init__.py │ │ ├── clip_score.py │ │ ├── cmmd │ │ │ ├── __init__.py │ │ │ ├── cmmd.py │ │ │ └── model.py │ │ ├── fid │ │ │ ├── __init__.py │ │ │ ├── compute_fid.py │ │ │ ├── fid.py │ │ │ ├── inception.py │ │ │ └── precision_recall.py │ │ ├── image_reward.py │ │ ├── inception_score │ │ │ ├── __init__.py │ │ │ └── inception_score.py │ │ ├── knn │ │ │ ├── __init__.py │ │ │ └── self_knn.py │ │ └── psnr │ │ │ ├── __init__.py │ │ │ └── psnr.py │ ├── trainer │ │ ├── __init__.py │ │ ├── dc_trainer.py │ │ ├── fsdp_wrap.py │ │ └── optimizer.py │ └── utils │ │ ├── __init__.py │ │ ├── activation_checkpointing.py │ │ ├── aspect_ratio.py │ │ ├── config.py │ │ ├── dist.py │ │ ├── efficiency.py │ │ ├── ema.py │ │ ├── export.py │ │ ├── image.py │ │ ├── init.py │ │ ├── lr.py │ │ ├── metric.py │ │ ├── prompt_safety_check.py │ │ ├── tar.py │ │ ├── tensorrt.py │ │ ├── video.py │ │ └── zip.py ├── c2i_model_zoo.py ├── c2icore │ ├── __init__.py │ ├── autoencoder.py │ ├── data_provider │ │ ├── __init__.py │ │ ├── base.py │ │ ├── latent_imagenet.py │ │ └── sample_class.py │ ├── diffusioncore │ │ ├── __init__.py │ │ ├── efficiency │ │ │ ├── __init__.py │ │ │ ├── onnx_export.py │ │ │ ├── test.py │ │ │ └── training.py │ │ ├── models │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── dit.py │ │ │ ├── dit_sampler │ │ │ │ ├── __init__.py │ │ │ │ ├── diffusion_utils.py │ │ │ │ ├── gaussian_diffusion.py │ │ │ │ ├── respace.py │ │ │ │ └── timestep_sampler.py │ │ │ ├── sana_cls.py │ │ │ ├── sana_utils │ │ │ │ ├── __init__.py │ │ │ │ ├── diffusion_utils.py │ │ │ │ ├── dpm_solver.py │ │ │ │ ├── gaussian_diffusion.py │ │ │ │ ├── respace.py │ │ │ │ ├── sana_block.py │ │ │ │ └── sana_sampler.py │ │ │ ├── sit_sampler │ │ │ │ ├── __init__.py │ │ │ │ ├── integrators.py │ │ │ │ ├── path.py │ │ │ │ ├── transport.py │ │ │ │ └── utils.py │ │ │ ├── uni_sampler │ │ │ │ ├── __init__.py │ │ │ │ ├── dpm_solver.py │ │ │ │ ├── model_wrapper.py │ │ │ │ ├── noise_schedule.py │ │ │ │ ├── sde_solver.py │ │ │ │ ├── uni_pc.py │ │ │ │ ├── uni_sampler.py │ │ │ │ └── uni_train_scheduler.py │ │ │ ├── uvit.py │ │ │ └── uvit_sampler │ │ │ │ ├── __init__.py │ │ │ │ ├── dpm_solver_pp.py │ │ │ │ └── uvit_schedule.py │ │ └── trainer.py │ ├── models │ │ ├── __init__.py │ │ ├── base.py │ │ └── ops │ │ │ ├── __init__.py │ │ │ ├── input_embed.py │ │ │ ├── label_embed.py │ │ │ └── timestep_embed.py │ └── trainer.py ├── models │ ├── __init__.py │ ├── nn │ │ ├── __init__.py │ │ ├── act.py │ │ ├── norm.py │ │ ├── ops.py │ │ └── triton_rms_norm.py │ └── utils │ │ ├── __init__.py │ │ ├── diff_aug.py │ │ ├── list.py │ │ ├── network.py │ │ ├── optimizer.py │ │ └── random.py └── t2icore │ ├── __init__.py │ ├── data_provider │ ├── __init__.py │ ├── base_eval.py │ ├── base_train.py │ ├── collection.py │ ├── latent_dummy.py │ ├── latent_mixture.py │ └── mjhq_text_prompt.py │ ├── metrics_offline │ ├── __init__.py │ └── compute_clip_score.py │ ├── models │ ├── __init__.py │ ├── base.py │ └── ops │ │ ├── __init__.py │ │ ├── caption_embed.py │ │ ├── input_embed.py │ │ └── timestep_embed.py │ ├── scheduler │ └── scm_scheduler.py │ ├── t2idiffusioncore │ ├── __init__.py │ ├── efficiency │ │ ├── model.py │ │ ├── onnx_export.py │ │ └── test.py │ ├── models │ │ ├── __init__.py │ │ ├── base.py │ │ ├── sana_blocks │ │ │ ├── __init__.py │ │ │ ├── attention.py │ │ │ ├── glumb.py │ │ │ ├── norm.py │ │ │ └── pos_embed.py │ │ ├── sana_sprint.py │ │ └── sana_t2i.py │ └── trainer.py │ ├── text_encoder.py │ └── trainer.py ├── projects ├── DC-AE-1.5.md ├── DC-AE-Lite.md └── DC-AE.md ├── pyproject.toml ├── requirements.txt ├── scripts └── data │ └── process_guided_diffusion_reference.py └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/README.md -------------------------------------------------------------------------------- /applications/dc_ae/dc_ae_generate_latent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/applications/dc_ae/dc_ae_generate_latent.py -------------------------------------------------------------------------------- /applications/dc_ae/demo_dc_ae_diffusion_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/applications/dc_ae/demo_dc_ae_diffusion_model.py -------------------------------------------------------------------------------- /applications/dc_ae/demo_dc_ae_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/applications/dc_ae/demo_dc_ae_model.py -------------------------------------------------------------------------------- /applications/dc_ae/demo_dc_ae_model_diffusers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/applications/dc_ae/demo_dc_ae_model_diffusers.py -------------------------------------------------------------------------------- /applications/dc_ae/eval_dc_ae_diffusion_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/applications/dc_ae/eval_dc_ae_diffusion_model.py -------------------------------------------------------------------------------- /applications/dc_ae/eval_dc_ae_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/applications/dc_ae/eval_dc_ae_model.py -------------------------------------------------------------------------------- /applications/dc_ae/generate_reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/applications/dc_ae/generate_reference.py -------------------------------------------------------------------------------- /applications/dc_ae/train_dc_ae_diffusion_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/applications/dc_ae/train_dc_ae_diffusion_model.py -------------------------------------------------------------------------------- /applications/dc_t2i/get_null_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/applications/dc_t2i/get_null_embedding.py -------------------------------------------------------------------------------- /applications/dc_t2i/manual_prompt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/applications/dc_t2i/manual_prompt.py -------------------------------------------------------------------------------- /assets/dc-gen-figures/analysis.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/assets/dc-gen-figures/analysis.jpg -------------------------------------------------------------------------------- /assets/dc-gen-figures/appendix_1K.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/assets/dc-gen-figures/appendix_1K.png -------------------------------------------------------------------------------- /assets/dc-gen-figures/comp_qualitative.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/assets/dc-gen-figures/comp_qualitative.jpg -------------------------------------------------------------------------------- /assets/dc-gen-figures/comp_speed.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/assets/dc-gen-figures/comp_speed.jpg -------------------------------------------------------------------------------- /assets/dc-gen-figures/pipeline.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/assets/dc-gen-figures/pipeline.jpg -------------------------------------------------------------------------------- /assets/dc-gen-figures/teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/assets/dc-gen-figures/teaser.jpg -------------------------------------------------------------------------------- /assets/docs/dc_ae_sana_1.1.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/assets/docs/dc_ae_sana_1.1.md -------------------------------------------------------------------------------- /assets/train_diffusion_with_flux_vae_f8.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/assets/train_diffusion_with_flux_vae_f8.md -------------------------------------------------------------------------------- /assets/train_diffusion_with_sd_vae_f8.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/assets/train_diffusion_with_sd_vae_f8.md -------------------------------------------------------------------------------- /dc_gen/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/ae_model_zoo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/ae_model_zoo.py -------------------------------------------------------------------------------- /dc_gen/aecore/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/aecore/data_provider/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/aecore/data_provider/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/aecore/data_provider/base.py -------------------------------------------------------------------------------- /dc_gen/aecore/data_provider/collection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/aecore/data_provider/collection.py -------------------------------------------------------------------------------- /dc_gen/aecore/data_provider/examine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/aecore/data_provider/examine.py -------------------------------------------------------------------------------- /dc_gen/aecore/data_provider/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/aecore/data_provider/imagenet.py -------------------------------------------------------------------------------- /dc_gen/aecore/data_provider/mixture.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/aecore/data_provider/mixture.py -------------------------------------------------------------------------------- /dc_gen/aecore/data_provider/mjhq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/aecore/data_provider/mjhq.py -------------------------------------------------------------------------------- /dc_gen/aecore/data_provider/vlm_caption.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/aecore/data_provider/vlm_caption.py -------------------------------------------------------------------------------- /dc_gen/aecore/efficiency/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/aecore/efficiency/test.py -------------------------------------------------------------------------------- /dc_gen/aecore/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/aecore/models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/aecore/models/base.py -------------------------------------------------------------------------------- /dc_gen/aecore/models/dc_ae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/aecore/models/dc_ae.py -------------------------------------------------------------------------------- /dc_gen/aecore/models/dc_ae_diffusers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/aecore/models/dc_ae_diffusers.py -------------------------------------------------------------------------------- /dc_gen/aecore/models/hf_autoencoder_kl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/aecore/models/hf_autoencoder_kl.py -------------------------------------------------------------------------------- /dc_gen/aecore/models/sd_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/aecore/models/sd_vae.py -------------------------------------------------------------------------------- /dc_gen/aecore/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/aecore/trainer.py -------------------------------------------------------------------------------- /dc_gen/apps/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/apps/data_provider/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/apps/data_provider/dc_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/data_provider/dc_base.py -------------------------------------------------------------------------------- /dc_gen/apps/data_provider/dc_mixture.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/data_provider/dc_mixture.py -------------------------------------------------------------------------------- /dc_gen/apps/data_provider/generate_archives.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/data_provider/generate_archives.py -------------------------------------------------------------------------------- /dc_gen/apps/data_provider/generate_latent_archives.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/data_provider/generate_latent_archives.py -------------------------------------------------------------------------------- /dc_gen/apps/data_provider/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/data_provider/sampler.py -------------------------------------------------------------------------------- /dc_gen/apps/data_provider/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/data_provider/utils.py -------------------------------------------------------------------------------- /dc_gen/apps/data_provider/web_dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/data_provider/web_dataset/__init__.py -------------------------------------------------------------------------------- /dc_gen/apps/data_provider/web_dataset/generate_meta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/data_provider/web_dataset/generate_meta.py -------------------------------------------------------------------------------- /dc_gen/apps/data_provider/web_dataset/wids.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/data_provider/web_dataset/wids.py -------------------------------------------------------------------------------- /dc_gen/apps/data_provider/web_dataset/wids_dl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/data_provider/web_dataset/wids_dl.py -------------------------------------------------------------------------------- /dc_gen/apps/data_provider/web_dataset/wids_lru.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/data_provider/web_dataset/wids_lru.py -------------------------------------------------------------------------------- /dc_gen/apps/data_provider/web_dataset/wids_mmtar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/data_provider/web_dataset/wids_mmtar.py -------------------------------------------------------------------------------- /dc_gen/apps/data_provider/web_dataset/wids_specs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/data_provider/web_dataset/wids_specs.py -------------------------------------------------------------------------------- /dc_gen/apps/data_provider/web_dataset/wids_tar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/data_provider/web_dataset/wids_tar.py -------------------------------------------------------------------------------- /dc_gen/apps/data_provider/zip_dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/apps/data_provider/zip_dataset/generate_meta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/data_provider/zip_dataset/generate_meta.py -------------------------------------------------------------------------------- /dc_gen/apps/data_provider/zip_dataset/single_zip_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/data_provider/zip_dataset/single_zip_dataset.py -------------------------------------------------------------------------------- /dc_gen/apps/data_provider/zip_dataset/zip_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/data_provider/zip_dataset/zip_dataset.py -------------------------------------------------------------------------------- /dc_gen/apps/metrics/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/apps/metrics/clip_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/metrics/clip_score.py -------------------------------------------------------------------------------- /dc_gen/apps/metrics/cmmd/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/apps/metrics/cmmd/cmmd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/metrics/cmmd/cmmd.py -------------------------------------------------------------------------------- /dc_gen/apps/metrics/cmmd/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/metrics/cmmd/model.py -------------------------------------------------------------------------------- /dc_gen/apps/metrics/fid/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/apps/metrics/fid/compute_fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/metrics/fid/compute_fid.py -------------------------------------------------------------------------------- /dc_gen/apps/metrics/fid/fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/metrics/fid/fid.py -------------------------------------------------------------------------------- /dc_gen/apps/metrics/fid/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/metrics/fid/inception.py -------------------------------------------------------------------------------- /dc_gen/apps/metrics/fid/precision_recall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/metrics/fid/precision_recall.py -------------------------------------------------------------------------------- /dc_gen/apps/metrics/image_reward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/metrics/image_reward.py -------------------------------------------------------------------------------- /dc_gen/apps/metrics/inception_score/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/apps/metrics/inception_score/inception_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/metrics/inception_score/inception_score.py -------------------------------------------------------------------------------- /dc_gen/apps/metrics/knn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/apps/metrics/knn/self_knn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/metrics/knn/self_knn.py -------------------------------------------------------------------------------- /dc_gen/apps/metrics/psnr/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/apps/metrics/psnr/psnr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/metrics/psnr/psnr.py -------------------------------------------------------------------------------- /dc_gen/apps/trainer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/apps/trainer/dc_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/trainer/dc_trainer.py -------------------------------------------------------------------------------- /dc_gen/apps/trainer/fsdp_wrap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/trainer/fsdp_wrap.py -------------------------------------------------------------------------------- /dc_gen/apps/trainer/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/trainer/optimizer.py -------------------------------------------------------------------------------- /dc_gen/apps/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/utils/__init__.py -------------------------------------------------------------------------------- /dc_gen/apps/utils/activation_checkpointing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/utils/activation_checkpointing.py -------------------------------------------------------------------------------- /dc_gen/apps/utils/aspect_ratio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/utils/aspect_ratio.py -------------------------------------------------------------------------------- /dc_gen/apps/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/utils/config.py -------------------------------------------------------------------------------- /dc_gen/apps/utils/dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/utils/dist.py -------------------------------------------------------------------------------- /dc_gen/apps/utils/efficiency.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/utils/efficiency.py -------------------------------------------------------------------------------- /dc_gen/apps/utils/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/utils/ema.py -------------------------------------------------------------------------------- /dc_gen/apps/utils/export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/utils/export.py -------------------------------------------------------------------------------- /dc_gen/apps/utils/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/utils/image.py -------------------------------------------------------------------------------- /dc_gen/apps/utils/init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/utils/init.py -------------------------------------------------------------------------------- /dc_gen/apps/utils/lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/utils/lr.py -------------------------------------------------------------------------------- /dc_gen/apps/utils/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/utils/metric.py -------------------------------------------------------------------------------- /dc_gen/apps/utils/prompt_safety_check.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/utils/prompt_safety_check.py -------------------------------------------------------------------------------- /dc_gen/apps/utils/tar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/utils/tar.py -------------------------------------------------------------------------------- /dc_gen/apps/utils/tensorrt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/utils/tensorrt.py -------------------------------------------------------------------------------- /dc_gen/apps/utils/video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/utils/video.py -------------------------------------------------------------------------------- /dc_gen/apps/utils/zip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/apps/utils/zip.py -------------------------------------------------------------------------------- /dc_gen/c2i_model_zoo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2i_model_zoo.py -------------------------------------------------------------------------------- /dc_gen/c2icore/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/c2icore/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/autoencoder.py -------------------------------------------------------------------------------- /dc_gen/c2icore/data_provider/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/c2icore/data_provider/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/data_provider/base.py -------------------------------------------------------------------------------- /dc_gen/c2icore/data_provider/latent_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/data_provider/latent_imagenet.py -------------------------------------------------------------------------------- /dc_gen/c2icore/data_provider/sample_class.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/data_provider/sample_class.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/efficiency/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/efficiency/onnx_export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/efficiency/onnx_export.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/efficiency/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/efficiency/test.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/efficiency/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/efficiency/training.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/base.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/dit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/dit.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/dit_sampler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/dit_sampler/__init__.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/dit_sampler/diffusion_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/dit_sampler/diffusion_utils.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/dit_sampler/gaussian_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/dit_sampler/gaussian_diffusion.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/dit_sampler/respace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/dit_sampler/respace.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/dit_sampler/timestep_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/dit_sampler/timestep_sampler.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/sana_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/sana_cls.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/sana_utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/sana_utils/diffusion_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/sana_utils/diffusion_utils.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/sana_utils/dpm_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/sana_utils/dpm_solver.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/sana_utils/gaussian_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/sana_utils/gaussian_diffusion.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/sana_utils/respace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/sana_utils/respace.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/sana_utils/sana_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/sana_utils/sana_block.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/sana_utils/sana_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/sana_utils/sana_sampler.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/sit_sampler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/sit_sampler/__init__.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/sit_sampler/integrators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/sit_sampler/integrators.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/sit_sampler/path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/sit_sampler/path.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/sit_sampler/transport.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/sit_sampler/transport.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/sit_sampler/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/sit_sampler/utils.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/uni_sampler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/uni_sampler/__init__.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/uni_sampler/dpm_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/uni_sampler/dpm_solver.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/uni_sampler/model_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/uni_sampler/model_wrapper.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/uni_sampler/noise_schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/uni_sampler/noise_schedule.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/uni_sampler/sde_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/uni_sampler/sde_solver.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/uni_sampler/uni_pc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/uni_sampler/uni_pc.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/uni_sampler/uni_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/uni_sampler/uni_sampler.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/uni_sampler/uni_train_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/uni_sampler/uni_train_scheduler.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/uvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/uvit.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/uvit_sampler/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/uvit_sampler/dpm_solver_pp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/uvit_sampler/dpm_solver_pp.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/models/uvit_sampler/uvit_schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/models/uvit_sampler/uvit_schedule.py -------------------------------------------------------------------------------- /dc_gen/c2icore/diffusioncore/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/diffusioncore/trainer.py -------------------------------------------------------------------------------- /dc_gen/c2icore/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/c2icore/models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/models/base.py -------------------------------------------------------------------------------- /dc_gen/c2icore/models/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/c2icore/models/ops/input_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/models/ops/input_embed.py -------------------------------------------------------------------------------- /dc_gen/c2icore/models/ops/label_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/models/ops/label_embed.py -------------------------------------------------------------------------------- /dc_gen/c2icore/models/ops/timestep_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/models/ops/timestep_embed.py -------------------------------------------------------------------------------- /dc_gen/c2icore/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/c2icore/trainer.py -------------------------------------------------------------------------------- /dc_gen/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/models/nn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/models/nn/__init__.py -------------------------------------------------------------------------------- /dc_gen/models/nn/act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/models/nn/act.py -------------------------------------------------------------------------------- /dc_gen/models/nn/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/models/nn/norm.py -------------------------------------------------------------------------------- /dc_gen/models/nn/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/models/nn/ops.py -------------------------------------------------------------------------------- /dc_gen/models/nn/triton_rms_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/models/nn/triton_rms_norm.py -------------------------------------------------------------------------------- /dc_gen/models/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/models/utils/__init__.py -------------------------------------------------------------------------------- /dc_gen/models/utils/diff_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/models/utils/diff_aug.py -------------------------------------------------------------------------------- /dc_gen/models/utils/list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/models/utils/list.py -------------------------------------------------------------------------------- /dc_gen/models/utils/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/models/utils/network.py -------------------------------------------------------------------------------- /dc_gen/models/utils/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/models/utils/optimizer.py -------------------------------------------------------------------------------- /dc_gen/models/utils/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/models/utils/random.py -------------------------------------------------------------------------------- /dc_gen/t2icore/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/t2icore/data_provider/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/t2icore/data_provider/base_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/t2icore/data_provider/base_eval.py -------------------------------------------------------------------------------- /dc_gen/t2icore/data_provider/base_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/t2icore/data_provider/base_train.py -------------------------------------------------------------------------------- /dc_gen/t2icore/data_provider/collection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/t2icore/data_provider/collection.py -------------------------------------------------------------------------------- /dc_gen/t2icore/data_provider/latent_dummy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/t2icore/data_provider/latent_dummy.py -------------------------------------------------------------------------------- /dc_gen/t2icore/data_provider/latent_mixture.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/t2icore/data_provider/latent_mixture.py -------------------------------------------------------------------------------- /dc_gen/t2icore/data_provider/mjhq_text_prompt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/t2icore/data_provider/mjhq_text_prompt.py -------------------------------------------------------------------------------- /dc_gen/t2icore/metrics_offline/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/t2icore/metrics_offline/compute_clip_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/t2icore/metrics_offline/compute_clip_score.py -------------------------------------------------------------------------------- /dc_gen/t2icore/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/t2icore/models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/t2icore/models/base.py -------------------------------------------------------------------------------- /dc_gen/t2icore/models/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/t2icore/models/ops/caption_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/t2icore/models/ops/caption_embed.py -------------------------------------------------------------------------------- /dc_gen/t2icore/models/ops/input_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/t2icore/models/ops/input_embed.py -------------------------------------------------------------------------------- /dc_gen/t2icore/models/ops/timestep_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/t2icore/models/ops/timestep_embed.py -------------------------------------------------------------------------------- /dc_gen/t2icore/scheduler/scm_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/t2icore/scheduler/scm_scheduler.py -------------------------------------------------------------------------------- /dc_gen/t2icore/t2idiffusioncore/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/t2icore/t2idiffusioncore/efficiency/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/t2icore/t2idiffusioncore/efficiency/model.py -------------------------------------------------------------------------------- /dc_gen/t2icore/t2idiffusioncore/efficiency/onnx_export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/t2icore/t2idiffusioncore/efficiency/onnx_export.py -------------------------------------------------------------------------------- /dc_gen/t2icore/t2idiffusioncore/efficiency/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/t2icore/t2idiffusioncore/efficiency/test.py -------------------------------------------------------------------------------- /dc_gen/t2icore/t2idiffusioncore/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/t2icore/t2idiffusioncore/models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/t2icore/t2idiffusioncore/models/base.py -------------------------------------------------------------------------------- /dc_gen/t2icore/t2idiffusioncore/models/sana_blocks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dc_gen/t2icore/t2idiffusioncore/models/sana_blocks/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/t2icore/t2idiffusioncore/models/sana_blocks/attention.py -------------------------------------------------------------------------------- /dc_gen/t2icore/t2idiffusioncore/models/sana_blocks/glumb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/t2icore/t2idiffusioncore/models/sana_blocks/glumb.py -------------------------------------------------------------------------------- /dc_gen/t2icore/t2idiffusioncore/models/sana_blocks/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/t2icore/t2idiffusioncore/models/sana_blocks/norm.py -------------------------------------------------------------------------------- /dc_gen/t2icore/t2idiffusioncore/models/sana_blocks/pos_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/t2icore/t2idiffusioncore/models/sana_blocks/pos_embed.py -------------------------------------------------------------------------------- /dc_gen/t2icore/t2idiffusioncore/models/sana_sprint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/t2icore/t2idiffusioncore/models/sana_sprint.py -------------------------------------------------------------------------------- /dc_gen/t2icore/t2idiffusioncore/models/sana_t2i.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/t2icore/t2idiffusioncore/models/sana_t2i.py -------------------------------------------------------------------------------- /dc_gen/t2icore/t2idiffusioncore/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/t2icore/t2idiffusioncore/trainer.py -------------------------------------------------------------------------------- /dc_gen/t2icore/text_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/t2icore/text_encoder.py -------------------------------------------------------------------------------- /dc_gen/t2icore/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/dc_gen/t2icore/trainer.py -------------------------------------------------------------------------------- /projects/DC-AE-1.5.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/projects/DC-AE-1.5.md -------------------------------------------------------------------------------- /projects/DC-AE-Lite.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/projects/DC-AE-Lite.md -------------------------------------------------------------------------------- /projects/DC-AE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/projects/DC-AE.md -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/data/process_guided_diffusion_reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/scripts/data/process_guided_diffusion_reference.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dc-ai-projects/DC-Gen/HEAD/setup.py --------------------------------------------------------------------------------