├── Detailed_instructions.md ├── LICENSE ├── README.md ├── assets └── images │ ├── pipeline_1col.png │ ├── qual_grid.png │ └── teaser_1col_v2.png ├── autoregressive ├── .DS_Store ├── lin_probe │ └── lin_probe.py ├── models │ ├── generate.py │ ├── gpt.py │ └── gpt_1d.py ├── sample │ ├── sample_c2i_cfg_search.py │ └── sample_c2i_ddp_visualization.py ├── serve │ ├── README.md │ ├── fake_json │ │ ├── GPT-3B.json │ │ ├── GPT-B.json │ │ ├── GPT-L.json │ │ ├── GPT-XL.json │ │ └── GPT-XXL.json │ ├── gpt_model.py │ ├── gpu_executor.py │ ├── llm.py │ ├── llm_engine.py │ ├── model_runner.py │ ├── sample_c2i.py │ ├── sampler.py │ └── worker.py └── train │ ├── check_codes_c2i.py │ ├── extract_codes_c2i.py │ ├── train_c2i.py │ ├── train_c2i_fsdp.py │ └── val_loss_cal.py ├── configs └── vq │ ├── VQ_BL256.yaml │ ├── VQ_BL256_2d.yaml │ ├── VQ_BL256_dino_disc.yaml │ ├── VQ_BL256_train_new.yaml │ ├── VQ_SB256.yaml │ ├── VQ_SB256_2d.yaml │ ├── VQ_SS256.yaml │ ├── VQ_SS256_2d.yaml │ └── VQ_XLXXL256.yaml ├── dataset ├── augmentation.py ├── build.py ├── coco.py ├── imagenet.py ├── openimage.py ├── pexels.py └── t2i.py ├── env_install.sh ├── evaluations └── c2i │ └── evaluator.py ├── scripts ├── check_code_c2i.sh ├── composite_cmd │ ├── eval_c2i.sh │ ├── train_all_lin_probe_and_eval.sh │ ├── train_c2i_and_eval.sh │ ├── train_c2i_and_eval_fsdp.sh │ ├── train_lm_lin_probe_and_eval.sh │ ├── train_tok_lin_probe_and_eval.sh │ ├── train_vq_and_eval.sh │ └── train_vq_and_eval_fsdp.sh ├── extract_code_c2i.sh ├── install_dependencies.sh ├── reconstruction.sh ├── sample_c2i_search_cfg.sh ├── sample_c2i_visualization.sh ├── train_c2i.sh ├── train_c2i_fsdp.sh ├── train_lin_probe.sh ├── train_lin_probe_gpt.sh ├── train_vq.sh ├── train_vq_fsdp.sh ├── val.sh └── val_loss_c2i.sh ├── set_env_vars.sh ├── tests └── gt_qualitiative │ ├── .DS_Store │ ├── 000001_gt.png │ ├── 000002_gt.png │ ├── 000003_gt.png │ ├── 000004_gt.png │ ├── 000005_gt.png │ ├── 000007_gt.png │ ├── 000008_gt.png │ └── 000009_gt.png ├── tokenizer ├── lin_probe │ ├── lin_probe.py │ └── misc.py ├── tokenizer_image │ ├── cache │ │ └── vgg.pth │ ├── diffaug.py │ ├── discriminator_dino.py │ ├── discriminator_patchgan.py │ ├── discriminator_patchgan_SeD.py │ ├── discriminator_patchr3gan.py │ ├── discriminator_stylegan.py │ ├── dist.py │ ├── fsq │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-39.pyc │ │ │ └── fsq_model.cpython-39.pyc │ │ └── fsq_model.py │ ├── lpips.py │ ├── lq │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-39.pyc │ │ │ └── lq_model.cpython-39.pyc │ │ ├── lq_loss.py │ │ ├── lq_model.py │ │ ├── lq_train.py │ │ └── lq_vit_model.py │ ├── module_attention_SeD.py │ ├── reconstruction_ddp.py │ ├── scheduler.py │ ├── vae │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-39.pyc │ │ │ └── vae_vit_model.cpython-39.pyc │ │ ├── reconstruction_vae_ddp.py │ │ ├── vae_loss.py │ │ ├── vae_train.py │ │ └── vae_vit_model.py │ └── vq │ │ ├── __init__.py │ │ ├── blocks.py │ │ ├── gptc.py │ │ ├── lfq.py │ │ ├── vq_loss.py │ │ ├── vq_train.py │ │ ├── vq_train_fsdp.py │ │ └── vq_vit_model.py ├── vae │ ├── README.md │ ├── reconstruction_vae_ddp.py │ └── sd_vae_demo.py └── validation │ └── val_ddp.py └── utils ├── data.py ├── deepspeed.py ├── distributed.py ├── drop_path.py ├── ema.py ├── imgnet_idx2label.py ├── logger.py ├── model_init.py ├── resume_log.py └── rope.py /Detailed_instructions.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/Detailed_instructions.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/README.md -------------------------------------------------------------------------------- /assets/images/pipeline_1col.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/assets/images/pipeline_1col.png -------------------------------------------------------------------------------- /assets/images/qual_grid.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/assets/images/qual_grid.png -------------------------------------------------------------------------------- /assets/images/teaser_1col_v2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/assets/images/teaser_1col_v2.png -------------------------------------------------------------------------------- /autoregressive/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/autoregressive/.DS_Store -------------------------------------------------------------------------------- /autoregressive/lin_probe/lin_probe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/autoregressive/lin_probe/lin_probe.py -------------------------------------------------------------------------------- /autoregressive/models/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/autoregressive/models/generate.py -------------------------------------------------------------------------------- /autoregressive/models/gpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/autoregressive/models/gpt.py -------------------------------------------------------------------------------- /autoregressive/models/gpt_1d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/autoregressive/models/gpt_1d.py -------------------------------------------------------------------------------- /autoregressive/sample/sample_c2i_cfg_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/autoregressive/sample/sample_c2i_cfg_search.py -------------------------------------------------------------------------------- /autoregressive/sample/sample_c2i_ddp_visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/autoregressive/sample/sample_c2i_ddp_visualization.py -------------------------------------------------------------------------------- /autoregressive/serve/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/autoregressive/serve/README.md -------------------------------------------------------------------------------- /autoregressive/serve/fake_json/GPT-3B.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/autoregressive/serve/fake_json/GPT-3B.json -------------------------------------------------------------------------------- /autoregressive/serve/fake_json/GPT-B.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/autoregressive/serve/fake_json/GPT-B.json -------------------------------------------------------------------------------- /autoregressive/serve/fake_json/GPT-L.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/autoregressive/serve/fake_json/GPT-L.json -------------------------------------------------------------------------------- /autoregressive/serve/fake_json/GPT-XL.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/autoregressive/serve/fake_json/GPT-XL.json -------------------------------------------------------------------------------- /autoregressive/serve/fake_json/GPT-XXL.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/autoregressive/serve/fake_json/GPT-XXL.json -------------------------------------------------------------------------------- /autoregressive/serve/gpt_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/autoregressive/serve/gpt_model.py -------------------------------------------------------------------------------- /autoregressive/serve/gpu_executor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/autoregressive/serve/gpu_executor.py -------------------------------------------------------------------------------- /autoregressive/serve/llm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/autoregressive/serve/llm.py -------------------------------------------------------------------------------- /autoregressive/serve/llm_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/autoregressive/serve/llm_engine.py -------------------------------------------------------------------------------- /autoregressive/serve/model_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/autoregressive/serve/model_runner.py -------------------------------------------------------------------------------- /autoregressive/serve/sample_c2i.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/autoregressive/serve/sample_c2i.py -------------------------------------------------------------------------------- /autoregressive/serve/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/autoregressive/serve/sampler.py -------------------------------------------------------------------------------- /autoregressive/serve/worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/autoregressive/serve/worker.py -------------------------------------------------------------------------------- /autoregressive/train/check_codes_c2i.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/autoregressive/train/check_codes_c2i.py -------------------------------------------------------------------------------- /autoregressive/train/extract_codes_c2i.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/autoregressive/train/extract_codes_c2i.py -------------------------------------------------------------------------------- /autoregressive/train/train_c2i.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/autoregressive/train/train_c2i.py -------------------------------------------------------------------------------- /autoregressive/train/train_c2i_fsdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/autoregressive/train/train_c2i_fsdp.py -------------------------------------------------------------------------------- /autoregressive/train/val_loss_cal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/autoregressive/train/val_loss_cal.py -------------------------------------------------------------------------------- /configs/vq/VQ_BL256.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/configs/vq/VQ_BL256.yaml -------------------------------------------------------------------------------- /configs/vq/VQ_BL256_2d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/configs/vq/VQ_BL256_2d.yaml -------------------------------------------------------------------------------- /configs/vq/VQ_BL256_dino_disc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/configs/vq/VQ_BL256_dino_disc.yaml -------------------------------------------------------------------------------- /configs/vq/VQ_BL256_train_new.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/configs/vq/VQ_BL256_train_new.yaml -------------------------------------------------------------------------------- /configs/vq/VQ_SB256.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/configs/vq/VQ_SB256.yaml -------------------------------------------------------------------------------- /configs/vq/VQ_SB256_2d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/configs/vq/VQ_SB256_2d.yaml -------------------------------------------------------------------------------- /configs/vq/VQ_SS256.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/configs/vq/VQ_SS256.yaml -------------------------------------------------------------------------------- /configs/vq/VQ_SS256_2d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/configs/vq/VQ_SS256_2d.yaml -------------------------------------------------------------------------------- /configs/vq/VQ_XLXXL256.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/configs/vq/VQ_XLXXL256.yaml -------------------------------------------------------------------------------- /dataset/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/dataset/augmentation.py -------------------------------------------------------------------------------- /dataset/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/dataset/build.py -------------------------------------------------------------------------------- /dataset/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/dataset/coco.py -------------------------------------------------------------------------------- /dataset/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/dataset/imagenet.py -------------------------------------------------------------------------------- /dataset/openimage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/dataset/openimage.py -------------------------------------------------------------------------------- /dataset/pexels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/dataset/pexels.py -------------------------------------------------------------------------------- /dataset/t2i.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/dataset/t2i.py -------------------------------------------------------------------------------- /env_install.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/env_install.sh -------------------------------------------------------------------------------- /evaluations/c2i/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/evaluations/c2i/evaluator.py -------------------------------------------------------------------------------- /scripts/check_code_c2i.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/scripts/check_code_c2i.sh -------------------------------------------------------------------------------- /scripts/composite_cmd/eval_c2i.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/scripts/composite_cmd/eval_c2i.sh -------------------------------------------------------------------------------- /scripts/composite_cmd/train_all_lin_probe_and_eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/scripts/composite_cmd/train_all_lin_probe_and_eval.sh -------------------------------------------------------------------------------- /scripts/composite_cmd/train_c2i_and_eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/scripts/composite_cmd/train_c2i_and_eval.sh -------------------------------------------------------------------------------- /scripts/composite_cmd/train_c2i_and_eval_fsdp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/scripts/composite_cmd/train_c2i_and_eval_fsdp.sh -------------------------------------------------------------------------------- /scripts/composite_cmd/train_lm_lin_probe_and_eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/scripts/composite_cmd/train_lm_lin_probe_and_eval.sh -------------------------------------------------------------------------------- /scripts/composite_cmd/train_tok_lin_probe_and_eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/scripts/composite_cmd/train_tok_lin_probe_and_eval.sh -------------------------------------------------------------------------------- /scripts/composite_cmd/train_vq_and_eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/scripts/composite_cmd/train_vq_and_eval.sh -------------------------------------------------------------------------------- /scripts/composite_cmd/train_vq_and_eval_fsdp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/scripts/composite_cmd/train_vq_and_eval_fsdp.sh -------------------------------------------------------------------------------- /scripts/extract_code_c2i.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/scripts/extract_code_c2i.sh -------------------------------------------------------------------------------- /scripts/install_dependencies.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/scripts/install_dependencies.sh -------------------------------------------------------------------------------- /scripts/reconstruction.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/scripts/reconstruction.sh -------------------------------------------------------------------------------- /scripts/sample_c2i_search_cfg.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/scripts/sample_c2i_search_cfg.sh -------------------------------------------------------------------------------- /scripts/sample_c2i_visualization.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/scripts/sample_c2i_visualization.sh -------------------------------------------------------------------------------- /scripts/train_c2i.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/scripts/train_c2i.sh -------------------------------------------------------------------------------- /scripts/train_c2i_fsdp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/scripts/train_c2i_fsdp.sh -------------------------------------------------------------------------------- /scripts/train_lin_probe.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/scripts/train_lin_probe.sh -------------------------------------------------------------------------------- /scripts/train_lin_probe_gpt.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/scripts/train_lin_probe_gpt.sh -------------------------------------------------------------------------------- /scripts/train_vq.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/scripts/train_vq.sh -------------------------------------------------------------------------------- /scripts/train_vq_fsdp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/scripts/train_vq_fsdp.sh -------------------------------------------------------------------------------- /scripts/val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/scripts/val.sh -------------------------------------------------------------------------------- /scripts/val_loss_c2i.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/scripts/val_loss_c2i.sh -------------------------------------------------------------------------------- /set_env_vars.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/set_env_vars.sh -------------------------------------------------------------------------------- /tests/gt_qualitiative/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tests/gt_qualitiative/.DS_Store -------------------------------------------------------------------------------- /tests/gt_qualitiative/000001_gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tests/gt_qualitiative/000001_gt.png -------------------------------------------------------------------------------- /tests/gt_qualitiative/000002_gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tests/gt_qualitiative/000002_gt.png -------------------------------------------------------------------------------- /tests/gt_qualitiative/000003_gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tests/gt_qualitiative/000003_gt.png -------------------------------------------------------------------------------- /tests/gt_qualitiative/000004_gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tests/gt_qualitiative/000004_gt.png -------------------------------------------------------------------------------- /tests/gt_qualitiative/000005_gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tests/gt_qualitiative/000005_gt.png -------------------------------------------------------------------------------- /tests/gt_qualitiative/000007_gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tests/gt_qualitiative/000007_gt.png -------------------------------------------------------------------------------- /tests/gt_qualitiative/000008_gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tests/gt_qualitiative/000008_gt.png -------------------------------------------------------------------------------- /tests/gt_qualitiative/000009_gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tests/gt_qualitiative/000009_gt.png -------------------------------------------------------------------------------- /tokenizer/lin_probe/lin_probe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/lin_probe/lin_probe.py -------------------------------------------------------------------------------- /tokenizer/lin_probe/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/lin_probe/misc.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/cache/vgg.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/cache/vgg.pth -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/diffaug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/diffaug.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/discriminator_dino.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/discriminator_dino.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/discriminator_patchgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/discriminator_patchgan.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/discriminator_patchgan_SeD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/discriminator_patchgan_SeD.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/discriminator_patchr3gan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/discriminator_patchr3gan.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/discriminator_stylegan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/discriminator_stylegan.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/dist.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/fsq/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/fsq/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/fsq/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/fsq/__pycache__/fsq_model.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/fsq/__pycache__/fsq_model.cpython-39.pyc -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/fsq/fsq_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/fsq/fsq_model.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/lpips.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/lq/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/lq/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/lq/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/lq/__pycache__/lq_model.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/lq/__pycache__/lq_model.cpython-39.pyc -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/lq/lq_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/lq/lq_loss.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/lq/lq_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/lq/lq_model.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/lq/lq_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/lq/lq_train.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/lq/lq_vit_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/lq/lq_vit_model.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/module_attention_SeD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/module_attention_SeD.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/reconstruction_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/reconstruction_ddp.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/scheduler.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/vae/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/vae/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/vae/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/vae/__pycache__/vae_vit_model.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/vae/__pycache__/vae_vit_model.cpython-39.pyc -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/vae/reconstruction_vae_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/vae/reconstruction_vae_ddp.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/vae/vae_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/vae/vae_loss.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/vae/vae_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/vae/vae_train.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/vae/vae_vit_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/vae/vae_vit_model.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/vq/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/vq/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/vq/blocks.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/vq/gptc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/vq/gptc.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/vq/lfq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/vq/lfq.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/vq/vq_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/vq/vq_loss.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/vq/vq_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/vq/vq_train.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/vq/vq_train_fsdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/vq/vq_train_fsdp.py -------------------------------------------------------------------------------- /tokenizer/tokenizer_image/vq/vq_vit_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/tokenizer_image/vq/vq_vit_model.py -------------------------------------------------------------------------------- /tokenizer/vae/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/vae/README.md -------------------------------------------------------------------------------- /tokenizer/vae/reconstruction_vae_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/vae/reconstruction_vae_ddp.py -------------------------------------------------------------------------------- /tokenizer/vae/sd_vae_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/vae/sd_vae_demo.py -------------------------------------------------------------------------------- /tokenizer/validation/val_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/tokenizer/validation/val_ddp.py -------------------------------------------------------------------------------- /utils/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/utils/data.py -------------------------------------------------------------------------------- /utils/deepspeed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/utils/deepspeed.py -------------------------------------------------------------------------------- /utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/utils/distributed.py -------------------------------------------------------------------------------- /utils/drop_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/utils/drop_path.py -------------------------------------------------------------------------------- /utils/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/utils/ema.py -------------------------------------------------------------------------------- /utils/imgnet_idx2label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/utils/imgnet_idx2label.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/model_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/utils/model_init.py -------------------------------------------------------------------------------- /utils/resume_log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/utils/resume_log.py -------------------------------------------------------------------------------- /utils/rope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SilentView/GigaTok/HEAD/utils/rope.py --------------------------------------------------------------------------------