├── .gitignore ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── Dockerfile ├── FAQ.md ├── LICENSE ├── MODEL_CARD.md ├── README.md ├── download.sh ├── example.py ├── llama ├── __init__.py ├── generation.py ├── model.py └── tokenizer.py ├── requirements.txt └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/modular-ml/wrapyfi-examples_llama/HEAD/.gitignore -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/modular-ml/wrapyfi-examples_llama/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/modular-ml/wrapyfi-examples_llama/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/modular-ml/wrapyfi-examples_llama/HEAD/Dockerfile -------------------------------------------------------------------------------- /FAQ.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/modular-ml/wrapyfi-examples_llama/HEAD/FAQ.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/modular-ml/wrapyfi-examples_llama/HEAD/LICENSE -------------------------------------------------------------------------------- /MODEL_CARD.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/modular-ml/wrapyfi-examples_llama/HEAD/MODEL_CARD.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/modular-ml/wrapyfi-examples_llama/HEAD/README.md -------------------------------------------------------------------------------- /download.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/modular-ml/wrapyfi-examples_llama/HEAD/download.sh -------------------------------------------------------------------------------- /example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/modular-ml/wrapyfi-examples_llama/HEAD/example.py -------------------------------------------------------------------------------- /llama/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/modular-ml/wrapyfi-examples_llama/HEAD/llama/__init__.py -------------------------------------------------------------------------------- /llama/generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/modular-ml/wrapyfi-examples_llama/HEAD/llama/generation.py -------------------------------------------------------------------------------- /llama/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/modular-ml/wrapyfi-examples_llama/HEAD/llama/model.py -------------------------------------------------------------------------------- /llama/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/modular-ml/wrapyfi-examples_llama/HEAD/llama/tokenizer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch 2 | fairscale 3 | fire 4 | sentencepiece -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/modular-ml/wrapyfi-examples_llama/HEAD/setup.py --------------------------------------------------------------------------------