path	# lines of code
benchmark/model_quality.py	115
benchmark/muse_chart.py	83
benchmark/muse_perf.py	433
benchmark/muse_table.py	62
muse/modeling_transformer_v2.py	740
muse/modeling_maskgit_vqgan.py	263
muse/logging.py	153
muse/modeling_paella_vq.py	149
muse/training_utils.py	233
muse/sampling.py	52
muse/modeling_utils.py	672
muse/lr_schedulers.py	103
muse/modeling_movq.py	460
muse/__init__.py	9
muse/modeling_taming_vqgan.py	417
muse/modeling_ema.py	137
muse/pipeline_muse.py	409
muse/modeling_transformer.py	1173
muse/data.py	1
pyproject.toml	5
training/train_muse.py	1055
training/optimizer.py	34
training/train_maskgit_imagenet.py	427
training/__init__.py	1
training/data.py	499
training/train_vqgan.py	1
scripts/convert_coco_to_wds.py	25
scripts/calculate_fid.py	193
scripts/gen_sdxl_synthetic_dataset.py	135
scripts/compute_offline_ema.py	42
scripts/convert_maskgit_vqgan.py	170
scripts/benchmark_models.py	44
scripts/log_inpainting_images.py	80
scripts/pre_encode.py	350
scripts/log_generations_wandb.py	65
scripts/convert_imagenet_to_wds.py	41
setup.py	42
configs/laion-aesthetic-475-max-1024-joined-with-stability-metadata-laicov2_shards.yaml	19
configs/cc12m.yaml	98
configs/cc12m_uvit_larger_paellavq_f8_clip.yaml	102
configs/cc12m_uvit.yaml	102
configs/research_run_512.yaml	120
configs/m4_shards.yaml	198
configs/cc12m_uvit_clip.yaml	102
configs/template_config.yaml	78
configs/research_run_512_with_downsample_finetune_on_sdxl_synthetic_data.yaml	119
configs/cc12m_movq.yaml	100
configs/imagenet_movq.yaml	84
configs/laiona6plus_uvit_clip_pool_embeds.yaml	103
configs/coyo700m_uvit_clip.yaml	100
configs/laiona6plus_uvit_clip_f8.yaml	101
configs/research_run_512_with_downsample.yaml	121
configs/laiona6plus_uvit_clip.yaml	100
configs/sdxl_synthetic_dataset_shards.yaml	127
configs/laiona6plus_prencode_random_replace.yaml	103
configs/laiona6plus_f8_preencoded.yaml	105
configs/cc12m_uvit_paellavq.yaml	103
configs/laion5plus_uvit_clip.yaml	100
configs/cc12m_uvit_paellavq_larger.yaml	102
configs/coyo_f8_preencoded.yaml	112
configs/imagenet.yaml	84
configs/coyo_f8_clip_embeds.yaml	113
configs/imagenet_text2image_movq_conv.yaml	99
configs/imagenet_text2image.yaml	95
setup.cfg	17