├── .gitignore ├── CITATION.cff ├── LICENSE ├── Makefile ├── README.md ├── benchmark ├── README.md ├── artifacts │ ├── 4090_bs_1.png │ ├── 4090_bs_8.png │ ├── 4090_old.png │ ├── a100_bs_1.png │ ├── a100_bs_8.png │ ├── a100_old.png │ ├── all.csv │ ├── clip.png │ ├── fid.png │ └── isc.png ├── model_quality.py ├── muse_chart.py ├── muse_perf.py └── muse_table.py ├── configs ├── cc12m.yaml ├── cc12m_movq.yaml ├── cc12m_uvit.yaml ├── cc12m_uvit_clip.yaml ├── cc12m_uvit_larger_paellavq_f8_clip.yaml ├── cc12m_uvit_paellavq.yaml ├── cc12m_uvit_paellavq_larger.yaml ├── coyo700m_uvit_clip.yaml ├── coyo_f8_clip_embeds.yaml ├── coyo_f8_preencoded.yaml ├── imagenet.yaml ├── imagenet_movq.yaml ├── imagenet_text2image.yaml ├── imagenet_text2image_movq_conv.yaml ├── laion-aesthetic-475-max-1024-joined-with-stability-metadata-laicov2_shards.yaml ├── laion5plus_uvit_clip.yaml ├── laiona6plus_f8_preencoded.yaml ├── laiona6plus_preencod_test.yaml ├── laiona6plus_prencode_random_replace.yaml ├── laiona6plus_uvit_clip.yaml ├── laiona6plus_uvit_clip_f8.yaml ├── laiona6plus_uvit_clip_pool_embeds.yaml ├── m4_shards.yaml ├── research_run_512.yaml ├── research_run_512_with_downsample.yaml ├── research_run_512_with_downsample_finetune_on_sdxl_synthetic_data.yaml ├── sdxl_synthetic_dataset_shards.yaml └── template_config.yaml ├── inpainting_validation ├── a bright jungle │ ├── image.jpeg │ └── mask.png ├── a field in the mountains │ ├── image.jpg │ └── mask.png ├── a man with glasses │ ├── image.jpeg │ └── mask.png ├── dog sitting on bench │ ├── image.jpg │ └── mask.png └── person sitting on bench outside │ ├── image.jpeg │ └── mask.png ├── muse ├── __init__.py ├── data.py ├── logging.py ├── lr_schedulers.py ├── modeling_ema.py ├── modeling_maskgit_vqgan.py ├── modeling_movq.py ├── modeling_paella_vq.py ├── modeling_taming_vqgan.py ├── modeling_transformer.py ├── modeling_transformer_v2.py ├── modeling_utils.py ├── pipeline_muse.py ├── sampling.py └── training_utils.py ├── pyproject.toml ├── scripts ├── benchmark_models.py ├── calculate_fid.py ├── compute_offline_ema.py ├── convert_coco_to_wds.py ├── convert_imagenet_to_wds.py ├── convert_maskgit_vqgan.py ├── gen_sdxl_synthetic_dataset.py ├── log_generations_wandb.py ├── log_inpainting_images.py └── pre_encode.py ├── setup.cfg ├── setup.py ├── slurm_scripts ├── calculate_fid_coco_calculate_fid.slurm ├── calculate_fid_coco_generate_images.slurm ├── cc12m.slurm ├── gen_sdxl_synthetic_dataset.slurm ├── imagenet.slurm ├── imagenet_movq_ds.slurm ├── imagenet_text2image.slurm ├── pre_encode_coyo.slurm ├── pre_encode_laion_5.slurm ├── pre_encode_laion_6.slurm ├── research_run_512.slurm ├── research_run_512_with_downsample.slurm └── research_run_512_with_downsample_finetune_on_sdxl_synthetic_data.slurm ├── test.py ├── training ├── __init__.py ├── data.py ├── optimizer.py ├── train_maskgit_imagenet.py ├── train_muse.py └── train_vqgan.py └── validation_prompts ├── cc_validation_prompts.txt └── dalle_mini_prompts.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/.gitignore -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/CITATION.cff -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/README.md -------------------------------------------------------------------------------- /benchmark/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/benchmark/README.md -------------------------------------------------------------------------------- /benchmark/artifacts/4090_bs_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/benchmark/artifacts/4090_bs_1.png -------------------------------------------------------------------------------- /benchmark/artifacts/4090_bs_8.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/benchmark/artifacts/4090_bs_8.png -------------------------------------------------------------------------------- /benchmark/artifacts/4090_old.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/benchmark/artifacts/4090_old.png -------------------------------------------------------------------------------- /benchmark/artifacts/a100_bs_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/benchmark/artifacts/a100_bs_1.png -------------------------------------------------------------------------------- /benchmark/artifacts/a100_bs_8.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/benchmark/artifacts/a100_bs_8.png -------------------------------------------------------------------------------- /benchmark/artifacts/a100_old.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/benchmark/artifacts/a100_old.png -------------------------------------------------------------------------------- /benchmark/artifacts/all.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/benchmark/artifacts/all.csv -------------------------------------------------------------------------------- /benchmark/artifacts/clip.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/benchmark/artifacts/clip.png -------------------------------------------------------------------------------- /benchmark/artifacts/fid.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/benchmark/artifacts/fid.png -------------------------------------------------------------------------------- /benchmark/artifacts/isc.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/benchmark/artifacts/isc.png -------------------------------------------------------------------------------- /benchmark/model_quality.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/benchmark/model_quality.py -------------------------------------------------------------------------------- /benchmark/muse_chart.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/benchmark/muse_chart.py -------------------------------------------------------------------------------- /benchmark/muse_perf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/benchmark/muse_perf.py -------------------------------------------------------------------------------- /benchmark/muse_table.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/benchmark/muse_table.py -------------------------------------------------------------------------------- /configs/cc12m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/cc12m.yaml -------------------------------------------------------------------------------- /configs/cc12m_movq.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/cc12m_movq.yaml -------------------------------------------------------------------------------- /configs/cc12m_uvit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/cc12m_uvit.yaml -------------------------------------------------------------------------------- /configs/cc12m_uvit_clip.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/cc12m_uvit_clip.yaml -------------------------------------------------------------------------------- /configs/cc12m_uvit_larger_paellavq_f8_clip.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/cc12m_uvit_larger_paellavq_f8_clip.yaml -------------------------------------------------------------------------------- /configs/cc12m_uvit_paellavq.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/cc12m_uvit_paellavq.yaml -------------------------------------------------------------------------------- /configs/cc12m_uvit_paellavq_larger.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/cc12m_uvit_paellavq_larger.yaml -------------------------------------------------------------------------------- /configs/coyo700m_uvit_clip.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/coyo700m_uvit_clip.yaml -------------------------------------------------------------------------------- /configs/coyo_f8_clip_embeds.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/coyo_f8_clip_embeds.yaml -------------------------------------------------------------------------------- /configs/coyo_f8_preencoded.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/coyo_f8_preencoded.yaml -------------------------------------------------------------------------------- /configs/imagenet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/imagenet.yaml -------------------------------------------------------------------------------- /configs/imagenet_movq.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/imagenet_movq.yaml -------------------------------------------------------------------------------- /configs/imagenet_text2image.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/imagenet_text2image.yaml -------------------------------------------------------------------------------- /configs/imagenet_text2image_movq_conv.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/imagenet_text2image_movq_conv.yaml -------------------------------------------------------------------------------- /configs/laion-aesthetic-475-max-1024-joined-with-stability-metadata-laicov2_shards.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/laion-aesthetic-475-max-1024-joined-with-stability-metadata-laicov2_shards.yaml -------------------------------------------------------------------------------- /configs/laion5plus_uvit_clip.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/laion5plus_uvit_clip.yaml -------------------------------------------------------------------------------- /configs/laiona6plus_f8_preencoded.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/laiona6plus_f8_preencoded.yaml -------------------------------------------------------------------------------- /configs/laiona6plus_preencod_test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/laiona6plus_preencod_test.yaml -------------------------------------------------------------------------------- /configs/laiona6plus_prencode_random_replace.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/laiona6plus_prencode_random_replace.yaml -------------------------------------------------------------------------------- /configs/laiona6plus_uvit_clip.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/laiona6plus_uvit_clip.yaml -------------------------------------------------------------------------------- /configs/laiona6plus_uvit_clip_f8.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/laiona6plus_uvit_clip_f8.yaml -------------------------------------------------------------------------------- /configs/laiona6plus_uvit_clip_pool_embeds.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/laiona6plus_uvit_clip_pool_embeds.yaml -------------------------------------------------------------------------------- /configs/m4_shards.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/m4_shards.yaml -------------------------------------------------------------------------------- /configs/research_run_512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/research_run_512.yaml -------------------------------------------------------------------------------- /configs/research_run_512_with_downsample.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/research_run_512_with_downsample.yaml -------------------------------------------------------------------------------- /configs/research_run_512_with_downsample_finetune_on_sdxl_synthetic_data.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/research_run_512_with_downsample_finetune_on_sdxl_synthetic_data.yaml -------------------------------------------------------------------------------- /configs/sdxl_synthetic_dataset_shards.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/sdxl_synthetic_dataset_shards.yaml -------------------------------------------------------------------------------- /configs/template_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/configs/template_config.yaml -------------------------------------------------------------------------------- /inpainting_validation/a bright jungle/image.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/inpainting_validation/a bright jungle/image.jpeg -------------------------------------------------------------------------------- /inpainting_validation/a bright jungle/mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/inpainting_validation/a bright jungle/mask.png -------------------------------------------------------------------------------- /inpainting_validation/a field in the mountains/image.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/inpainting_validation/a field in the mountains/image.jpg -------------------------------------------------------------------------------- /inpainting_validation/a field in the mountains/mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/inpainting_validation/a field in the mountains/mask.png -------------------------------------------------------------------------------- /inpainting_validation/a man with glasses/image.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/inpainting_validation/a man with glasses/image.jpeg -------------------------------------------------------------------------------- /inpainting_validation/a man with glasses/mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/inpainting_validation/a man with glasses/mask.png -------------------------------------------------------------------------------- /inpainting_validation/dog sitting on bench/image.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/inpainting_validation/dog sitting on bench/image.jpg -------------------------------------------------------------------------------- /inpainting_validation/dog sitting on bench/mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/inpainting_validation/dog sitting on bench/mask.png -------------------------------------------------------------------------------- /inpainting_validation/person sitting on bench outside/image.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/inpainting_validation/person sitting on bench outside/image.jpeg -------------------------------------------------------------------------------- /inpainting_validation/person sitting on bench outside/mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/inpainting_validation/person sitting on bench outside/mask.png -------------------------------------------------------------------------------- /muse/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/muse/__init__.py -------------------------------------------------------------------------------- /muse/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/muse/data.py -------------------------------------------------------------------------------- /muse/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/muse/logging.py -------------------------------------------------------------------------------- /muse/lr_schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/muse/lr_schedulers.py -------------------------------------------------------------------------------- /muse/modeling_ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/muse/modeling_ema.py -------------------------------------------------------------------------------- /muse/modeling_maskgit_vqgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/muse/modeling_maskgit_vqgan.py -------------------------------------------------------------------------------- /muse/modeling_movq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/muse/modeling_movq.py -------------------------------------------------------------------------------- /muse/modeling_paella_vq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/muse/modeling_paella_vq.py -------------------------------------------------------------------------------- /muse/modeling_taming_vqgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/muse/modeling_taming_vqgan.py -------------------------------------------------------------------------------- /muse/modeling_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/muse/modeling_transformer.py -------------------------------------------------------------------------------- /muse/modeling_transformer_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/muse/modeling_transformer_v2.py -------------------------------------------------------------------------------- /muse/modeling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/muse/modeling_utils.py -------------------------------------------------------------------------------- /muse/pipeline_muse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/muse/pipeline_muse.py -------------------------------------------------------------------------------- /muse/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/muse/sampling.py -------------------------------------------------------------------------------- /muse/training_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/muse/training_utils.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/pyproject.toml -------------------------------------------------------------------------------- /scripts/benchmark_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/scripts/benchmark_models.py -------------------------------------------------------------------------------- /scripts/calculate_fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/scripts/calculate_fid.py -------------------------------------------------------------------------------- /scripts/compute_offline_ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/scripts/compute_offline_ema.py -------------------------------------------------------------------------------- /scripts/convert_coco_to_wds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/scripts/convert_coco_to_wds.py -------------------------------------------------------------------------------- /scripts/convert_imagenet_to_wds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/scripts/convert_imagenet_to_wds.py -------------------------------------------------------------------------------- /scripts/convert_maskgit_vqgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/scripts/convert_maskgit_vqgan.py -------------------------------------------------------------------------------- /scripts/gen_sdxl_synthetic_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/scripts/gen_sdxl_synthetic_dataset.py -------------------------------------------------------------------------------- /scripts/log_generations_wandb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/scripts/log_generations_wandb.py -------------------------------------------------------------------------------- /scripts/log_inpainting_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/scripts/log_inpainting_images.py -------------------------------------------------------------------------------- /scripts/pre_encode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/scripts/pre_encode.py -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/setup.py -------------------------------------------------------------------------------- /slurm_scripts/calculate_fid_coco_calculate_fid.slurm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/slurm_scripts/calculate_fid_coco_calculate_fid.slurm -------------------------------------------------------------------------------- /slurm_scripts/calculate_fid_coco_generate_images.slurm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/slurm_scripts/calculate_fid_coco_generate_images.slurm -------------------------------------------------------------------------------- /slurm_scripts/cc12m.slurm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/slurm_scripts/cc12m.slurm -------------------------------------------------------------------------------- /slurm_scripts/gen_sdxl_synthetic_dataset.slurm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/slurm_scripts/gen_sdxl_synthetic_dataset.slurm -------------------------------------------------------------------------------- /slurm_scripts/imagenet.slurm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/slurm_scripts/imagenet.slurm -------------------------------------------------------------------------------- /slurm_scripts/imagenet_movq_ds.slurm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/slurm_scripts/imagenet_movq_ds.slurm -------------------------------------------------------------------------------- /slurm_scripts/imagenet_text2image.slurm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/slurm_scripts/imagenet_text2image.slurm -------------------------------------------------------------------------------- /slurm_scripts/pre_encode_coyo.slurm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/slurm_scripts/pre_encode_coyo.slurm -------------------------------------------------------------------------------- /slurm_scripts/pre_encode_laion_5.slurm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/slurm_scripts/pre_encode_laion_5.slurm -------------------------------------------------------------------------------- /slurm_scripts/pre_encode_laion_6.slurm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/slurm_scripts/pre_encode_laion_6.slurm -------------------------------------------------------------------------------- /slurm_scripts/research_run_512.slurm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/slurm_scripts/research_run_512.slurm -------------------------------------------------------------------------------- /slurm_scripts/research_run_512_with_downsample.slurm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/slurm_scripts/research_run_512_with_downsample.slurm -------------------------------------------------------------------------------- /slurm_scripts/research_run_512_with_downsample_finetune_on_sdxl_synthetic_data.slurm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/slurm_scripts/research_run_512_with_downsample_finetune_on_sdxl_synthetic_data.slurm -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/test.py -------------------------------------------------------------------------------- /training/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/training/__init__.py -------------------------------------------------------------------------------- /training/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/training/data.py -------------------------------------------------------------------------------- /training/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/training/optimizer.py -------------------------------------------------------------------------------- /training/train_maskgit_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/training/train_maskgit_imagenet.py -------------------------------------------------------------------------------- /training/train_muse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/training/train_muse.py -------------------------------------------------------------------------------- /training/train_vqgan.py: -------------------------------------------------------------------------------- 1 | """Training script for VQGAN.""" 2 | -------------------------------------------------------------------------------- /validation_prompts/cc_validation_prompts.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/validation_prompts/cc_validation_prompts.txt -------------------------------------------------------------------------------- /validation_prompts/dalle_mini_prompts.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/open-muse/HEAD/validation_prompts/dalle_mini_prompts.txt --------------------------------------------------------------------------------