├── .gitignore ├── LICENSE ├── README.md ├── main.py ├── mistral ├── __init__.py ├── cache.py ├── model.py ├── moe.py ├── rope.py └── tokenizer.py └── requirements.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkproj/mistral-src-commented/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkproj/mistral-src-commented/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkproj/mistral-src-commented/HEAD/README.md -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkproj/mistral-src-commented/HEAD/main.py -------------------------------------------------------------------------------- /mistral/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mistral/cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkproj/mistral-src-commented/HEAD/mistral/cache.py -------------------------------------------------------------------------------- /mistral/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkproj/mistral-src-commented/HEAD/mistral/model.py -------------------------------------------------------------------------------- /mistral/moe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkproj/mistral-src-commented/HEAD/mistral/moe.py -------------------------------------------------------------------------------- /mistral/rope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkproj/mistral-src-commented/HEAD/mistral/rope.py -------------------------------------------------------------------------------- /mistral/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkproj/mistral-src-commented/HEAD/mistral/tokenizer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | fire 2 | sentencepiece 3 | torch>=2.1.0 4 | xformers 5 | simple-parsing --------------------------------------------------------------------------------