├── .github └── workflows │ └── python-publish.yml ├── .gitignore ├── LICENSE ├── MANIFEST.in ├── README.md ├── nuwa.png ├── nuwa_pytorch ├── __init__.py ├── data │ └── bpe_simple_vocab_16e6.txt ├── image_utils.py ├── nuwa_pytorch.py ├── optimizer.py ├── reversible.py ├── reversible_video_audio.py ├── tokenizer.py ├── train_nuwa.py ├── train_vqgan_vae.py └── vqgan_vae.py └── setup.py /.github/workflows/python-publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/nuwa-pytorch/HEAD/.github/workflows/python-publish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/nuwa-pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/nuwa-pytorch/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- 1 | recursive-include nuwa_pytorch *.txt 2 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/nuwa-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /nuwa.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/nuwa-pytorch/HEAD/nuwa.png -------------------------------------------------------------------------------- /nuwa_pytorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/nuwa-pytorch/HEAD/nuwa_pytorch/__init__.py -------------------------------------------------------------------------------- /nuwa_pytorch/data/bpe_simple_vocab_16e6.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/nuwa-pytorch/HEAD/nuwa_pytorch/data/bpe_simple_vocab_16e6.txt -------------------------------------------------------------------------------- /nuwa_pytorch/image_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/nuwa-pytorch/HEAD/nuwa_pytorch/image_utils.py -------------------------------------------------------------------------------- /nuwa_pytorch/nuwa_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/nuwa-pytorch/HEAD/nuwa_pytorch/nuwa_pytorch.py -------------------------------------------------------------------------------- /nuwa_pytorch/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/nuwa-pytorch/HEAD/nuwa_pytorch/optimizer.py -------------------------------------------------------------------------------- /nuwa_pytorch/reversible.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/nuwa-pytorch/HEAD/nuwa_pytorch/reversible.py -------------------------------------------------------------------------------- /nuwa_pytorch/reversible_video_audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/nuwa-pytorch/HEAD/nuwa_pytorch/reversible_video_audio.py -------------------------------------------------------------------------------- /nuwa_pytorch/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/nuwa-pytorch/HEAD/nuwa_pytorch/tokenizer.py -------------------------------------------------------------------------------- /nuwa_pytorch/train_nuwa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/nuwa-pytorch/HEAD/nuwa_pytorch/train_nuwa.py -------------------------------------------------------------------------------- /nuwa_pytorch/train_vqgan_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/nuwa-pytorch/HEAD/nuwa_pytorch/train_vqgan_vae.py -------------------------------------------------------------------------------- /nuwa_pytorch/vqgan_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/nuwa-pytorch/HEAD/nuwa_pytorch/vqgan_vae.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/nuwa-pytorch/HEAD/setup.py --------------------------------------------------------------------------------