Initial commit
parents
No related branches found
No related tags found
Showing
- .gitignore 160 additions, 0 deletions.gitignore
- LICENSE 201 additions, 0 deletionsLICENSE
- README.md 96 additions, 0 deletionsREADME.md
- assets/attention_through_layers.png 0 additions, 0 deletionsassets/attention_through_layers.png
- assets/chunking.png 0 additions, 0 deletionsassets/chunking.png
- assets/full_attention.png 0 additions, 0 deletionsassets/full_attention.png
- assets/kv_padding.png 0 additions, 0 deletionsassets/kv_padding.png
- assets/padding.png 0 additions, 0 deletionsassets/padding.png
- assets/rolling_cache.png 0 additions, 0 deletionsassets/rolling_cache.png
- assets/sliding_attention.png 0 additions, 0 deletionsassets/sliding_attention.png
- main.py 137 additions, 0 deletionsmain.py
- mistral/__init__.py 0 additions, 0 deletionsmistral/__init__.py
- mistral/cache.py 213 additions, 0 deletionsmistral/cache.py
- mistral/model.py 223 additions, 0 deletionsmistral/model.py
- mistral/rope.py 22 additions, 0 deletionsmistral/rope.py
- mistral/tokenizer.py 36 additions, 0 deletionsmistral/tokenizer.py
- one_file_ref.py 366 additions, 0 deletionsone_file_ref.py
- requirements.txt 5 additions, 0 deletionsrequirements.txt
- test_generate.py 95 additions, 0 deletionstest_generate.py
Loading
Please register or sign in to comment