LICENSE
README.md
pyproject.toml
./medlat/__init__.py
./medlat/registry.py
./medlat/utils.py
./medlat/diffusion/__init__.py
./medlat/diffusion/diffloss.py
./medlat/diffusion/gaussian_diffusion.py
./medlat/diffusion/loss_weighting.py
./medlat/diffusion/losses.py
./medlat/diffusion/resample.py
./medlat/diffusion/respace.py
./medlat/first_stage/__init__.py
./medlat/first_stage/continuous/__init__.py
./medlat/first_stage/continuous/register.py
./medlat/first_stage/continuous/vae_models.py
./medlat/first_stage/continuous/modules/__init__.py
./medlat/first_stage/continuous/modules/ldm_modules.py
./medlat/first_stage/continuous/modules/maisi_modules.py
./medlat/first_stage/continuous/modules/dcae_modules/__init__.py
./medlat/first_stage/continuous/modules/dcae_modules/dcae_modules.py
./medlat/first_stage/continuous/modules/dcae_modules/efficientvit/__init__.py
./medlat/first_stage/continuous/modules/dcae_modules/efficientvit/backbone.py
./medlat/first_stage/continuous/modules/dcae_modules/efficientvit/cls.py
./medlat/first_stage/continuous/modules/dcae_modules/efficientvit/dc_ae.py
./medlat/first_stage/continuous/modules/dcae_modules/efficientvit/sam.py
./medlat/first_stage/continuous/modules/dcae_modules/efficientvit/seg.py
./medlat/first_stage/continuous/modules/dcae_modules/nn/__init__.py
./medlat/first_stage/continuous/modules/dcae_modules/nn/act.py
./medlat/first_stage/continuous/modules/dcae_modules/nn/drop.py
./medlat/first_stage/continuous/modules/dcae_modules/nn/norm.py
./medlat/first_stage/continuous/modules/dcae_modules/nn/ops.py
./medlat/first_stage/continuous/modules/dcae_modules/nn/triton_rms_norm.py
./medlat/first_stage/continuous/modules/dcae_modules/utils/__init__.py
./medlat/first_stage/continuous/modules/dcae_modules/utils/list.py
./medlat/first_stage/continuous/modules/dcae_modules/utils/network.py
./medlat/first_stage/continuous/modules/dcae_modules/utils/random.py
./medlat/first_stage/discrete/__init__.py
./medlat/first_stage/discrete/register.py
./medlat/first_stage/discrete/vq_models.py
./medlat/first_stage/discrete/modules/ldm_modules.py
./medlat/first_stage/discrete/quantizer/__init__.py
./medlat/first_stage/discrete/quantizer/modules.py
./medlat/first_stage/discrete/quantizer/quantize.py
./medlat/first_stage/modules/gaussian_dist.py
./medlat/first_stage/token/__init__.py
./medlat/first_stage/token/register.py
./medlat/first_stage/token/detok/__init__.py
./medlat/first_stage/token/detok/detok.py
./medlat/first_stage/token/maetok/__init__.py
./medlat/first_stage/token/maetok/maetok.py
./medlat/first_stage/token/maetok/modules/__init__.py
./medlat/first_stage/token/maetok/modules/hog.py
./medlat/first_stage/token/maetok/modules/vit_models.py
./medlat/first_stage/token/titok/__init__.py
./medlat/first_stage/token/titok/modules.py
./medlat/first_stage/token/titok/titok.py
./medlat/first_stage/token/vita/__init__.py
./medlat/first_stage/token/vita/modules.py
./medlat/first_stage/token/vita/vita.py
./medlat/first_stage/token/vita/utils/imaging_model_related.py
./medlat/first_stage/token/vmae/__init__.py
./medlat/first_stage/token/vmae/models.py
./medlat/first_stage/token/vmae/register.py
./medlat/generators/__init__.py
./medlat/generators/autoregressive/__init__.py
./medlat/generators/autoregressive/fractal/__init__.py
./medlat/generators/autoregressive/fractal/ar.py
./medlat/generators/autoregressive/fractal/fractalgen.py
./medlat/generators/autoregressive/fractal/mar.py
./medlat/generators/autoregressive/fractal/models.py
./medlat/generators/autoregressive/fractal/pixelloss.py
./medlat/generators/autoregressive/fractal/register.py
./medlat/generators/autoregressive/fractal/util.py
./medlat/generators/autoregressive/mage/__init__.py
./medlat/generators/autoregressive/mage/mage.py
./medlat/generators/autoregressive/mage/register.py
./medlat/generators/autoregressive/mage/vit_mage.py
./medlat/generators/autoregressive/mar/__init__.py
./medlat/generators/autoregressive/mar/mar.py
./medlat/generators/autoregressive/mar/register.py
./medlat/generators/autoregressive/maskbit/__init__.py
./medlat/generators/autoregressive/maskbit/base_model.py
./medlat/generators/autoregressive/maskbit/maskbit.py
./medlat/generators/autoregressive/maskbit/register.py
./medlat/generators/autoregressive/maskbit/util/__init__.py
./medlat/generators/autoregressive/maskbit/util/factorization.py
./medlat/generators/autoregressive/maskbit/util/masking.py
./medlat/generators/autoregressive/maskbit/util/pseudo_3d_sampling.py
./medlat/generators/autoregressive/maskbit/util/quantizer_utils.py
./medlat/generators/autoregressive/maskbit/util/sampling.py
./medlat/generators/autoregressive/maskgit/__init__.py
./medlat/generators/autoregressive/maskgit/layers.py
./medlat/generators/autoregressive/maskgit/register.py
./medlat/generators/autoregressive/maskgit/transformer.py
./medlat/generators/autoregressive/maskgit/txt_transformer.py
./medlat/generators/autoregressive/maskgit/utils/pos_embed.py
./medlat/generators/autoregressive/muse/__init__.py
./medlat/generators/autoregressive/muse/attend.py
./medlat/generators/autoregressive/muse/ldm_vq.py
./medlat/generators/autoregressive/muse/muse_maskgit.py
./medlat/generators/autoregressive/muse/t5.py
./medlat/generators/autoregressive/rar/__init__.py
./medlat/generators/autoregressive/rar/modules.py
./medlat/generators/autoregressive/rar/rar.py
./medlat/generators/autoregressive/rar/rardiff.py
./medlat/generators/autoregressive/rar/register.py
./medlat/generators/autoregressive/taming/__init__.py
./medlat/generators/autoregressive/taming/register.py
./medlat/generators/autoregressive/taming/transformer.py
./medlat/generators/autoregressive/var/helpers.py
./medlat/generators/autoregressive/var/modules.py
./medlat/generators/autoregressive/var/var.py
./medlat/generators/non_autoregressive/__init__.py
./medlat/generators/non_autoregressive/adm/__init__.py
./medlat/generators/non_autoregressive/adm/nn.py
./medlat/generators/non_autoregressive/adm/register.py
./medlat/generators/non_autoregressive/adm/unet.py
./medlat/generators/non_autoregressive/diffae/__init__.py
./medlat/generators/non_autoregressive/diffae/bg_blocks.py
./medlat/generators/non_autoregressive/diffae/bg_choices.py
./medlat/generators/non_autoregressive/diffae/bg_config_base.py
./medlat/generators/non_autoregressive/diffae/bg_latentnet.py
./medlat/generators/non_autoregressive/diffae/bg_nn.py
./medlat/generators/non_autoregressive/diffae/bg_unet.py
./medlat/generators/non_autoregressive/diffae/bg_unet_autoenc.py
./medlat/generators/non_autoregressive/dit/__init__.py
./medlat/generators/non_autoregressive/dit/models.py
./medlat/generators/non_autoregressive/dit/register.py
./medlat/generators/non_autoregressive/ldm/__init__.py
./medlat/generators/non_autoregressive/ldm/attention.py
./medlat/generators/non_autoregressive/ldm/nn.py
./medlat/generators/non_autoregressive/ldm/openaimodel.py
./medlat/generators/non_autoregressive/ldm/register.py
./medlat/generators/non_autoregressive/ldm/util.py
./medlat/generators/non_autoregressive/mdt/__init__.py
./medlat/generators/non_autoregressive/mdt/models.py
./medlat/generators/non_autoregressive/mdt/register.py
./medlat/generators/non_autoregressive/uvit/__init__.py
./medlat/generators/non_autoregressive/uvit/models.py
./medlat/generators/non_autoregressive/uvit/register.py
./medlat/modules/alignments.py
./medlat/modules/in_and_out.py
./medlat/modules/pos_embed.py
./medlat/modules/vit_core.py
./medlat/modules/wrapper.py
medlat/__init__.py
medlat/registry.py
medlat/utils.py
medlat.egg-info/PKG-INFO
medlat.egg-info/SOURCES.txt
medlat.egg-info/dependency_links.txt
medlat.egg-info/requires.txt
medlat.egg-info/top_level.txt
medlat/diffusion/__init__.py
medlat/diffusion/diffloss.py
medlat/diffusion/gaussian_diffusion.py
medlat/diffusion/loss_weighting.py
medlat/diffusion/losses.py
medlat/diffusion/resample.py
medlat/diffusion/respace.py
medlat/first_stage/__init__.py
medlat/first_stage/continuous/__init__.py
medlat/first_stage/continuous/register.py
medlat/first_stage/continuous/vae_models.py
medlat/first_stage/continuous/modules/__init__.py
medlat/first_stage/continuous/modules/ldm_modules.py
medlat/first_stage/continuous/modules/maisi_modules.py
medlat/first_stage/continuous/modules/dcae_modules/__init__.py
medlat/first_stage/continuous/modules/dcae_modules/dcae_modules.py
medlat/first_stage/continuous/modules/dcae_modules/efficientvit/__init__.py
medlat/first_stage/continuous/modules/dcae_modules/efficientvit/backbone.py
medlat/first_stage/continuous/modules/dcae_modules/efficientvit/cls.py
medlat/first_stage/continuous/modules/dcae_modules/efficientvit/dc_ae.py
medlat/first_stage/continuous/modules/dcae_modules/efficientvit/sam.py
medlat/first_stage/continuous/modules/dcae_modules/efficientvit/seg.py
medlat/first_stage/continuous/modules/dcae_modules/nn/__init__.py
medlat/first_stage/continuous/modules/dcae_modules/nn/act.py
medlat/first_stage/continuous/modules/dcae_modules/nn/drop.py
medlat/first_stage/continuous/modules/dcae_modules/nn/norm.py
medlat/first_stage/continuous/modules/dcae_modules/nn/ops.py
medlat/first_stage/continuous/modules/dcae_modules/nn/triton_rms_norm.py
medlat/first_stage/continuous/modules/dcae_modules/utils/__init__.py
medlat/first_stage/continuous/modules/dcae_modules/utils/list.py
medlat/first_stage/continuous/modules/dcae_modules/utils/network.py
medlat/first_stage/continuous/modules/dcae_modules/utils/random.py
medlat/first_stage/discrete/__init__.py
medlat/first_stage/discrete/register.py
medlat/first_stage/discrete/vq_models.py
medlat/first_stage/discrete/modules/ldm_modules.py
medlat/first_stage/discrete/quantizer/__init__.py
medlat/first_stage/discrete/quantizer/modules.py
medlat/first_stage/discrete/quantizer/quantize.py
medlat/first_stage/modules/gaussian_dist.py
medlat/first_stage/token/__init__.py
medlat/first_stage/token/register.py
medlat/first_stage/token/detok/__init__.py
medlat/first_stage/token/detok/detok.py
medlat/first_stage/token/maetok/__init__.py
medlat/first_stage/token/maetok/maetok.py
medlat/first_stage/token/maetok/modules/__init__.py
medlat/first_stage/token/maetok/modules/hog.py
medlat/first_stage/token/maetok/modules/vit_models.py
medlat/first_stage/token/titok/__init__.py
medlat/first_stage/token/titok/modules.py
medlat/first_stage/token/titok/titok.py
medlat/first_stage/token/vita/__init__.py
medlat/first_stage/token/vita/modules.py
medlat/first_stage/token/vita/vita.py
medlat/first_stage/token/vita/utils/imaging_model_related.py
medlat/first_stage/token/vmae/__init__.py
medlat/first_stage/token/vmae/models.py
medlat/first_stage/token/vmae/register.py
medlat/generators/__init__.py
medlat/generators/autoregressive/__init__.py
medlat/generators/autoregressive/mar/__init__.py
medlat/generators/autoregressive/mar/mar.py
medlat/generators/autoregressive/mar/register.py
medlat/generators/autoregressive/maskgit/__init__.py
medlat/generators/autoregressive/maskgit/layers.py
medlat/generators/autoregressive/maskgit/register.py
medlat/generators/autoregressive/maskgit/transformer.py
medlat/generators/autoregressive/maskgit/txt_transformer.py
medlat/generators/autoregressive/maskgit/utils/pos_embed.py
medlat/generators/non_autoregressive/__init__.py
medlat/generators/non_autoregressive/adm/__init__.py
medlat/generators/non_autoregressive/adm/nn.py
medlat/generators/non_autoregressive/adm/register.py
medlat/generators/non_autoregressive/adm/unet.py
medlat/generators/non_autoregressive/dit/__init__.py
medlat/generators/non_autoregressive/dit/models.py
medlat/generators/non_autoregressive/dit/register.py
medlat/generators/non_autoregressive/ldm/__init__.py
medlat/generators/non_autoregressive/ldm/attention.py
medlat/generators/non_autoregressive/ldm/nn.py
medlat/generators/non_autoregressive/ldm/openaimodel.py
medlat/generators/non_autoregressive/ldm/register.py
medlat/generators/non_autoregressive/ldm/util.py
medlat/modules/alignments.py
medlat/modules/in_and_out.py
medlat/modules/pos_embed.py
medlat/modules/vit_core.py
medlat/modules/wrapper.py
tests/test_smoke.py