Add the MPT model. (#1114)

* Add the MPT model.

* Add ffn and block.

* Forward pass for the mpt block.

* Repeat-kv.
This commit is contained in:
Laurent Mazare
2023-10-17 16:06:48 +01:00
committed by GitHub
parent f9e93f5b69
commit 872c3f14b0
2 changed files with 203 additions and 0 deletions

View File

@ -7,6 +7,7 @@ pub mod falcon;
pub mod llama;
pub mod mistral;
pub mod mixformer;
pub mod mpt;
pub mod quantized_llama;
pub mod quantized_mistral;
pub mod quantized_mixformer;