CHANGELOG.md
LICENSE
MANIFEST.in
README.md
pyproject.toml
requirements.txt
configs/model_scaling_variants.yaml
pytorch_image_translation_model.egg-info/PKG-INFO
pytorch_image_translation_model.egg-info/SOURCES.txt
pytorch_image_translation_model.egg-info/dependency_links.txt
pytorch_image_translation_model.egg-info/requires.txt
pytorch_image_translation_model.egg-info/top_level.txt
src/__init__.py
src/models/__init__.py
src/models/dit/__init__.py
src/models/dit/pixeldit.py
src/models/dit/pixnerd.py
src/models/dit/sit.py
src/models/gan/__init__.py
src/models/gan/cut_model.py
src/models/gan/stegogan_model.py
src/models/turbo/__init__.py
src/models/turbo/cyclegan_turbo.py
src/models/turbo/pix2pix_turbo.py
src/models/unet/__init__.py
src/models/unet/unet_2d.py
src/models/unet/unet_cdtsde.py
src/models/unet/unet_unidb.py
src/pipelines/__init__.py
src/pipelines/bdbm/__init__.py
src/pipelines/bdbm/pipeline_bdbm.py
src/pipelines/bdbm/pipeline_bdbm_latent.py
src/pipelines/bibbdm/__init__.py
src/pipelines/bibbdm/pipeline_bibbdm.py
src/pipelines/bibbdm/pipeline_bibbdm_latent.py
src/pipelines/cdtsde/__init__.py
src/pipelines/cdtsde/pipeline_cdtsde.py
src/pipelines/cdtsde/pipeline_cdtsde_latent.py
src/pipelines/cut/__init__.py
src/pipelines/cut/pipeline_cut.py
src/pipelines/cut/pipeline_cut_latent.py
src/pipelines/dab/__init__.py
src/pipelines/dab/pipeline_dab.py
src/pipelines/dab/pipeline_dab_latent.py
src/pipelines/dbim/__init__.py
src/pipelines/dbim/pipeline_dbim.py
src/pipelines/dbim/pipeline_dbim_latent.py
src/pipelines/ddbm/__init__.py
src/pipelines/ddbm/pipeline_ddbm.py
src/pipelines/ddbm/pipeline_ddbm_latent.py
src/pipelines/ddib/__init__.py
src/pipelines/ddib/pipeline_ddib.py
src/pipelines/ddib/pipeline_ddib_latent.py
src/pipelines/i2sb/__init__.py
src/pipelines/i2sb/pipeline_i2sb.py
src/pipelines/i2sb/pipeline_i2sb_latent.py
src/pipelines/sid/__init__.py
src/pipelines/sid/pipeline_sid.py
src/pipelines/sid2/__init__.py
src/pipelines/sid2/pipeline_sid2.py
src/pipelines/stegogan/__init__.py
src/pipelines/stegogan/pipeline_stegogan.py
src/pipelines/turbo/__init__.py
src/pipelines/turbo/pipeline_turbo.py
src/pipelines/unidb/__init__.py
src/pipelines/unidb/pipeline_unidb.py
src/schedulers/__init__.py
src/schedulers/scheduling_bdbm.py
src/schedulers/scheduling_bibbdm.py
src/schedulers/scheduling_cdtsde.py
src/schedulers/scheduling_cut.py
src/schedulers/scheduling_dab.py
src/schedulers/scheduling_dbim.py
src/schedulers/scheduling_ddbm.py
src/schedulers/scheduling_ddib.py
src/schedulers/scheduling_edm2.py
src/schedulers/scheduling_i2sb.py
src/schedulers/scheduling_sid.py
src/schedulers/scheduling_sid2.py
src/schedulers/scheduling_stegogan.py
src/schedulers/scheduling_turbo.py
src/schedulers/scheduling_unidb.py
src/schedulers/scheduling_vdm.py
src/utils/__init__.py
src/utils/cut_util.py
src/utils/dino_struct.py
src/utils/fp16_util.py
src/utils/fp16_util_ddib.py
src/utils/image_pool.py
src/utils/image_prep.py
src/utils/latent_target.py
src/utils/mavic_t_dataset.py
src/utils/metrics.py
src/utils/multidiffusion.py
src/utils/nn.py
src/utils/nn_ddib.py
src/utils/paths.py
src/utils/readme_utils.py
src/utils/rep_alignment.py
src/utils/training_utils.py
src/utils/turbo_training_utils.py
src/utils/vae_utils.py
tests/test_ablation_features.py
tests/test_bdbm_baseline.py
tests/test_bibbdm_baseline.py
tests/test_cdtsde_baseline.py
tests/test_checkpoint_defaults.py
tests/test_dab_baseline.py
tests/test_dbim_baseline.py
tests/test_ddbm_cfg_pixel_pipeline.py
tests/test_edm2_vdm_sid_schedulers.py
tests/test_efficient_attention.py
tests/test_ema_checkpoint.py
tests/test_filter_bad_samples.py
tests/test_from_pretrained.py
tests/test_i2sb_baseline.py
tests/test_latent_channels_config.py
tests/test_latent_pipelines.py
tests/test_log_validation.py
tests/test_metrics.py
tests/test_packaging.py
tests/test_pixeldit_backbone.py
tests/test_pixnerd_backbone.py
tests/test_push_checkpoint_to_hub.py
tests/test_runtime_random_crop.py
tests/test_shared_repa_latent.py
tests/test_sid2_baseline.py
tests/test_sit_backbone.py
tests/test_stegogan_baseline.py
tests/test_training_utils.py
tests/test_vae_reconstruction_script.py