├── .gitignore ├── .gitmodules ├── .isort.cfg ├── .pre-commit-config.yaml ├── LICENSES ├── AGPL-3.0-only.txt ├── CC-BY-4.0.txt ├── CC0-1.0.txt └── GPL-3.0-only.txt ├── README.md ├── environment.yml ├── media ├── console.gif └── console.gif.license ├── non-nlp ├── additional_tasks.py ├── generation_strategy.py ├── import_picoclvr.py ├── main.py ├── order.py ├── rejection_sampling.py └── sigmagpt.py └── text ├── caching.py ├── import_nanogpt.py ├── order.py ├── pytest.ini ├── sample.py ├── sigmagpt.py ├── tests └── test_flash_attention.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/.gitmodules -------------------------------------------------------------------------------- /.isort.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/.isort.cfg -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSES/AGPL-3.0-only.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/LICENSES/AGPL-3.0-only.txt -------------------------------------------------------------------------------- /LICENSES/CC-BY-4.0.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/LICENSES/CC-BY-4.0.txt -------------------------------------------------------------------------------- /LICENSES/CC0-1.0.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/LICENSES/CC0-1.0.txt -------------------------------------------------------------------------------- /LICENSES/GPL-3.0-only.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/LICENSES/GPL-3.0-only.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/README.md -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/environment.yml -------------------------------------------------------------------------------- /media/console.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/media/console.gif -------------------------------------------------------------------------------- /media/console.gif.license: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/media/console.gif.license -------------------------------------------------------------------------------- /non-nlp/additional_tasks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/non-nlp/additional_tasks.py -------------------------------------------------------------------------------- /non-nlp/generation_strategy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/non-nlp/generation_strategy.py -------------------------------------------------------------------------------- /non-nlp/import_picoclvr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/non-nlp/import_picoclvr.py -------------------------------------------------------------------------------- /non-nlp/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/non-nlp/main.py -------------------------------------------------------------------------------- /non-nlp/order.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/non-nlp/order.py -------------------------------------------------------------------------------- /non-nlp/rejection_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/non-nlp/rejection_sampling.py -------------------------------------------------------------------------------- /non-nlp/sigmagpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/non-nlp/sigmagpt.py -------------------------------------------------------------------------------- /text/caching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/text/caching.py -------------------------------------------------------------------------------- /text/import_nanogpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/text/import_nanogpt.py -------------------------------------------------------------------------------- /text/order.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/text/order.py -------------------------------------------------------------------------------- /text/pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/text/pytest.ini -------------------------------------------------------------------------------- /text/sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/text/sample.py -------------------------------------------------------------------------------- /text/sigmagpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/text/sigmagpt.py -------------------------------------------------------------------------------- /text/tests/test_flash_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/text/tests/test_flash_attention.py -------------------------------------------------------------------------------- /text/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/idiap/sigma-gpt/HEAD/text/train.py --------------------------------------------------------------------------------