├── .gitignore ├── README.hf.md ├── README.md ├── ae.py ├── challenge.ipynb ├── contents ├── appg.png ├── buildings.jpg ├── chinatown.jpg ├── cosplayers.jpg ├── flowers.jpg ├── lavender.jpg ├── logo.png ├── magazines.jpg ├── origin.png ├── randomwoman.jpeg ├── recon.png ├── ti2.png └── ti2_mask.png ├── inference.ipynb ├── launcher.sh ├── scripts └── launch_hdr.sh ├── sweep.sh ├── tae.py ├── tester.py ├── tester_upload.sh ├── unit_activation_reinitializer.py ├── utils.py └── vae_trainer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/.gitignore -------------------------------------------------------------------------------- /README.hf.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/README.hf.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/README.md -------------------------------------------------------------------------------- /ae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/ae.py -------------------------------------------------------------------------------- /challenge.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/challenge.ipynb -------------------------------------------------------------------------------- /contents/appg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/contents/appg.png -------------------------------------------------------------------------------- /contents/buildings.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/contents/buildings.jpg -------------------------------------------------------------------------------- /contents/chinatown.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/contents/chinatown.jpg -------------------------------------------------------------------------------- /contents/cosplayers.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/contents/cosplayers.jpg -------------------------------------------------------------------------------- /contents/flowers.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/contents/flowers.jpg -------------------------------------------------------------------------------- /contents/lavender.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/contents/lavender.jpg -------------------------------------------------------------------------------- /contents/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/contents/logo.png -------------------------------------------------------------------------------- /contents/magazines.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/contents/magazines.jpg -------------------------------------------------------------------------------- /contents/origin.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/contents/origin.png -------------------------------------------------------------------------------- /contents/randomwoman.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/contents/randomwoman.jpeg -------------------------------------------------------------------------------- /contents/recon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/contents/recon.png -------------------------------------------------------------------------------- /contents/ti2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/contents/ti2.png -------------------------------------------------------------------------------- /contents/ti2_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/contents/ti2_mask.png -------------------------------------------------------------------------------- /inference.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/inference.ipynb -------------------------------------------------------------------------------- /launcher.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/launcher.sh -------------------------------------------------------------------------------- /scripts/launch_hdr.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/scripts/launch_hdr.sh -------------------------------------------------------------------------------- /sweep.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/sweep.sh -------------------------------------------------------------------------------- /tae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/tae.py -------------------------------------------------------------------------------- /tester.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/tester.py -------------------------------------------------------------------------------- /tester_upload.sh: -------------------------------------------------------------------------------- 1 | export HF_TRANSFER=True 2 | huggingface-cli upload fal/AuraEquiVAE ./vae_epoch_3_step_49501_bf16.pt -------------------------------------------------------------------------------- /unit_activation_reinitializer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/unit_activation_reinitializer.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/utils.py -------------------------------------------------------------------------------- /vae_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/vqgan-training/HEAD/vae_trainer.py --------------------------------------------------------------------------------