├── .github └── workflows │ └── pypi-release.yml ├── .gitignore ├── CITATION.cff ├── LICENSE ├── README.md ├── img └── snac.png ├── requirements.txt ├── setup.py └── snac ├── __init__.py ├── attention.py ├── layers.py ├── snac.py └── vq.py /.github/workflows/pypi-release.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hubertsiuzdak/snac/HEAD/.github/workflows/pypi-release.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hubertsiuzdak/snac/HEAD/.gitignore -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hubertsiuzdak/snac/HEAD/CITATION.cff -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hubertsiuzdak/snac/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hubertsiuzdak/snac/HEAD/README.md -------------------------------------------------------------------------------- /img/snac.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hubertsiuzdak/snac/HEAD/img/snac.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch 2 | numpy 3 | einops 4 | huggingface_hub 5 | -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hubertsiuzdak/snac/HEAD/setup.py -------------------------------------------------------------------------------- /snac/__init__.py: -------------------------------------------------------------------------------- 1 | from .snac import SNAC 2 | 3 | __version__ = "1.2.1" 4 | -------------------------------------------------------------------------------- /snac/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hubertsiuzdak/snac/HEAD/snac/attention.py -------------------------------------------------------------------------------- /snac/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hubertsiuzdak/snac/HEAD/snac/layers.py -------------------------------------------------------------------------------- /snac/snac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hubertsiuzdak/snac/HEAD/snac/snac.py -------------------------------------------------------------------------------- /snac/vq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hubertsiuzdak/snac/HEAD/snac/vq.py --------------------------------------------------------------------------------