├── .github └── workflows │ └── python-publish.yml ├── .gitignore ├── LICENSE ├── README.md ├── magvit2.png ├── magvit2_pytorch ├── __init__.py ├── attend.py ├── data.py ├── magvit2_pytorch.py ├── optimizer.py ├── trainer.py └── version.py └── setup.py /.github/workflows/python-publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/magvit2-pytorch/HEAD/.github/workflows/python-publish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/magvit2-pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/magvit2-pytorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/magvit2-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /magvit2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/magvit2-pytorch/HEAD/magvit2.png -------------------------------------------------------------------------------- /magvit2_pytorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/magvit2-pytorch/HEAD/magvit2_pytorch/__init__.py -------------------------------------------------------------------------------- /magvit2_pytorch/attend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/magvit2-pytorch/HEAD/magvit2_pytorch/attend.py -------------------------------------------------------------------------------- /magvit2_pytorch/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/magvit2-pytorch/HEAD/magvit2_pytorch/data.py -------------------------------------------------------------------------------- /magvit2_pytorch/magvit2_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/magvit2-pytorch/HEAD/magvit2_pytorch/magvit2_pytorch.py -------------------------------------------------------------------------------- /magvit2_pytorch/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/magvit2-pytorch/HEAD/magvit2_pytorch/optimizer.py -------------------------------------------------------------------------------- /magvit2_pytorch/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/magvit2-pytorch/HEAD/magvit2_pytorch/trainer.py -------------------------------------------------------------------------------- /magvit2_pytorch/version.py: -------------------------------------------------------------------------------- 1 | __version__ = '0.5.1' 2 | -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/magvit2-pytorch/HEAD/setup.py --------------------------------------------------------------------------------