├── LICENSE ├── README.md ├── UViT_ImageNet_demo.ipynb ├── configs ├── celeba64_uvit_small.py ├── cifar10_uvit_small.py ├── imagenet256_uvit_huge.py ├── imagenet256_uvit_large.py ├── imagenet512_uvit_huge.py ├── imagenet512_uvit_large.py ├── imagenet64_uvit_large.py ├── imagenet64_uvit_mid.py └── mscoco_uvit_small.py ├── datasets.py ├── dpm_solver_pp.py ├── dpm_solver_pytorch.py ├── eval.py ├── eval_ldm.py ├── eval_ldm_discrete.py ├── eval_t2i_discrete.py ├── libs ├── __init__.py ├── autoencoder.py ├── clip.py ├── timm.py ├── uvit.py └── uvit_t2i.py ├── sample.png ├── sample_t2i_discrete.py ├── scripts ├── extract_empty_feature.py ├── extract_imagenet_feature.py ├── extract_mscoco_feature.py └── extract_test_prompt_feature.py ├── sde.py ├── skip_im.png ├── tools ├── fid_score.py └── inception.py ├── train.py ├── train_ldm.py ├── train_ldm_discrete.py ├── train_t2i_discrete.py ├── utils.py └── uvit.png /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/README.md -------------------------------------------------------------------------------- /UViT_ImageNet_demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/UViT_ImageNet_demo.ipynb -------------------------------------------------------------------------------- /configs/celeba64_uvit_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/configs/celeba64_uvit_small.py -------------------------------------------------------------------------------- /configs/cifar10_uvit_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/configs/cifar10_uvit_small.py -------------------------------------------------------------------------------- /configs/imagenet256_uvit_huge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/configs/imagenet256_uvit_huge.py -------------------------------------------------------------------------------- /configs/imagenet256_uvit_large.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/configs/imagenet256_uvit_large.py -------------------------------------------------------------------------------- /configs/imagenet512_uvit_huge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/configs/imagenet512_uvit_huge.py -------------------------------------------------------------------------------- /configs/imagenet512_uvit_large.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/configs/imagenet512_uvit_large.py -------------------------------------------------------------------------------- /configs/imagenet64_uvit_large.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/configs/imagenet64_uvit_large.py -------------------------------------------------------------------------------- /configs/imagenet64_uvit_mid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/configs/imagenet64_uvit_mid.py -------------------------------------------------------------------------------- /configs/mscoco_uvit_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/configs/mscoco_uvit_small.py -------------------------------------------------------------------------------- /datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/datasets.py -------------------------------------------------------------------------------- /dpm_solver_pp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/dpm_solver_pp.py -------------------------------------------------------------------------------- /dpm_solver_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/dpm_solver_pytorch.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/eval.py -------------------------------------------------------------------------------- /eval_ldm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/eval_ldm.py -------------------------------------------------------------------------------- /eval_ldm_discrete.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/eval_ldm_discrete.py -------------------------------------------------------------------------------- /eval_t2i_discrete.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/eval_t2i_discrete.py -------------------------------------------------------------------------------- /libs/__init__.py: -------------------------------------------------------------------------------- 1 | # codes from third party 2 | -------------------------------------------------------------------------------- /libs/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/libs/autoencoder.py -------------------------------------------------------------------------------- /libs/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/libs/clip.py -------------------------------------------------------------------------------- /libs/timm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/libs/timm.py -------------------------------------------------------------------------------- /libs/uvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/libs/uvit.py -------------------------------------------------------------------------------- /libs/uvit_t2i.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/libs/uvit_t2i.py -------------------------------------------------------------------------------- /sample.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/sample.png -------------------------------------------------------------------------------- /sample_t2i_discrete.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/sample_t2i_discrete.py -------------------------------------------------------------------------------- /scripts/extract_empty_feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/scripts/extract_empty_feature.py -------------------------------------------------------------------------------- /scripts/extract_imagenet_feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/scripts/extract_imagenet_feature.py -------------------------------------------------------------------------------- /scripts/extract_mscoco_feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/scripts/extract_mscoco_feature.py -------------------------------------------------------------------------------- /scripts/extract_test_prompt_feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/scripts/extract_test_prompt_feature.py -------------------------------------------------------------------------------- /sde.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/sde.py -------------------------------------------------------------------------------- /skip_im.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/skip_im.png -------------------------------------------------------------------------------- /tools/fid_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/tools/fid_score.py -------------------------------------------------------------------------------- /tools/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/tools/inception.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/train.py -------------------------------------------------------------------------------- /train_ldm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/train_ldm.py -------------------------------------------------------------------------------- /train_ldm_discrete.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/train_ldm_discrete.py -------------------------------------------------------------------------------- /train_t2i_discrete.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/train_t2i_discrete.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/utils.py -------------------------------------------------------------------------------- /uvit.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baofff/U-ViT/HEAD/uvit.png --------------------------------------------------------------------------------