add mamba
Showing
- README.md 2 additions, 2 deletionsREADME.md
- moe_one_file_ref.py 19 additions, 45 deletionsmoe_one_file_ref.py
- one_file_ref.py 17 additions, 39 deletionsone_file_ref.py
- src/mistral_inference/args.py 49 additions, 0 deletionssrc/mistral_inference/args.py
- src/mistral_inference/cache.py 15 additions, 38 deletionssrc/mistral_inference/cache.py
- src/mistral_inference/generate.py 38 additions, 15 deletionssrc/mistral_inference/generate.py
- src/mistral_inference/lora.py 23 additions, 34 deletionssrc/mistral_inference/lora.py
- src/mistral_inference/main.py 51 additions, 29 deletionssrc/mistral_inference/main.py
- src/mistral_inference/mamba.py 83 additions, 0 deletionssrc/mistral_inference/mamba.py
- src/mistral_inference/model.py 17 additions, 383 deletionssrc/mistral_inference/model.py
- src/mistral_inference/moe.py 2 additions, 6 deletionssrc/mistral_inference/moe.py
- src/mistral_inference/transformer.py 367 additions, 0 deletionssrc/mistral_inference/transformer.py
- tests/test_generate.py 36 additions, 22 deletionstests/test_generate.py
- tutorials/getting_started.ipynb 1 addition, 1 deletiontutorials/getting_started.ipynb
Loading
Please register or sign in to comment