├── .gitignore ├── CLIP ├── LICENSE.md └── clip │ ├── __init__.py │ ├── bpe_simple_vocab_16e6.txt.gz │ ├── clip_custom.py │ ├── model.py │ └── simple_tokenizer.py ├── LICENSE.md ├── README.md ├── assets └── fa-vae.png ├── cat_scripts ├── script_cat_celeba.sh └── train_cat.py ├── datasets ├── __init__.py ├── check_pkl_files.py ├── general_dataloader.py ├── general_dataloader_gpt.py ├── preprocess_celeba.py └── statistic.py ├── environment.yaml ├── favae_scripts ├── train_favae.py ├── train_favae_celeba.sh └── train_favae_other_datasets_public.sh ├── losses ├── hinge.py ├── inception.py ├── lpips.py └── vqgan_losses.py ├── models ├── .ipynb_checkpoints │ └── gpt-checkpoint.py ├── __init__.py ├── codec.py ├── discriminator.py ├── gpt_ca.py ├── l2_quantize.py ├── txt_cond_transformer.py └── vqgan_fcm.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/.gitignore -------------------------------------------------------------------------------- /CLIP/LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/CLIP/LICENSE.md -------------------------------------------------------------------------------- /CLIP/clip/__init__.py: -------------------------------------------------------------------------------- 1 | from .clip_custom import * 2 | -------------------------------------------------------------------------------- /CLIP/clip/bpe_simple_vocab_16e6.txt.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/CLIP/clip/bpe_simple_vocab_16e6.txt.gz -------------------------------------------------------------------------------- /CLIP/clip/clip_custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/CLIP/clip/clip_custom.py -------------------------------------------------------------------------------- /CLIP/clip/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/CLIP/clip/model.py -------------------------------------------------------------------------------- /CLIP/clip/simple_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/CLIP/clip/simple_tokenizer.py -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/README.md -------------------------------------------------------------------------------- /assets/fa-vae.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/assets/fa-vae.png -------------------------------------------------------------------------------- /cat_scripts/script_cat_celeba.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/cat_scripts/script_cat_celeba.sh -------------------------------------------------------------------------------- /cat_scripts/train_cat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/cat_scripts/train_cat.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/check_pkl_files.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/datasets/check_pkl_files.py -------------------------------------------------------------------------------- /datasets/general_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/datasets/general_dataloader.py -------------------------------------------------------------------------------- /datasets/general_dataloader_gpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/datasets/general_dataloader_gpt.py -------------------------------------------------------------------------------- /datasets/preprocess_celeba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/datasets/preprocess_celeba.py -------------------------------------------------------------------------------- /datasets/statistic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/datasets/statistic.py -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/environment.yaml -------------------------------------------------------------------------------- /favae_scripts/train_favae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/favae_scripts/train_favae.py -------------------------------------------------------------------------------- /favae_scripts/train_favae_celeba.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/favae_scripts/train_favae_celeba.sh -------------------------------------------------------------------------------- /favae_scripts/train_favae_other_datasets_public.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/favae_scripts/train_favae_other_datasets_public.sh -------------------------------------------------------------------------------- /losses/hinge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/losses/hinge.py -------------------------------------------------------------------------------- /losses/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/losses/inception.py -------------------------------------------------------------------------------- /losses/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/losses/lpips.py -------------------------------------------------------------------------------- /losses/vqgan_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/losses/vqgan_losses.py -------------------------------------------------------------------------------- /models/.ipynb_checkpoints/gpt-checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/models/.ipynb_checkpoints/gpt-checkpoint.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/codec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/models/codec.py -------------------------------------------------------------------------------- /models/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/models/discriminator.py -------------------------------------------------------------------------------- /models/gpt_ca.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/models/gpt_ca.py -------------------------------------------------------------------------------- /models/l2_quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/models/l2_quantize.py -------------------------------------------------------------------------------- /models/txt_cond_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/models/txt_cond_transformer.py -------------------------------------------------------------------------------- /models/vqgan_fcm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/models/vqgan_fcm.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oppo-us-research/FA-VAE/HEAD/utils.py --------------------------------------------------------------------------------