Skip to content

Commit

Permalink
Add the mixtral method
Browse files Browse the repository at this point in the history
  • Loading branch information
EricLBuehler committed Mar 6, 2024
1 parent 5dc1f22 commit 00ac1e1
Show file tree
Hide file tree
Showing 8 changed files with 1,002 additions and 5 deletions.
3 changes: 3 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,9 @@ Mistral.rs is a LLM inference platform written in pure, safe Rust.
- GGUF
- GGML
- X-LoRA
- Mixtral 8x7B
- Normal
- GGUF

**Library API**
- Rust multithreaded API for easy integration into any application: [docs](https://ericlbuehler.github.io/mistral.rs/mistralrs/). To use, add `mistralrs = { git = "https://github.com/EricLBuehler/mistral.rs.git" }` to the Cargo.toml.
Expand Down
3 changes: 2 additions & 1 deletion mistralrs-core/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,8 @@ mod xlora_models;

pub use pipeline::{
Conversation, GemmaLoader, GemmaSpecificConfig, LlamaLoader, LlamaSpecificConfig, Loader,
MistralLoader, MistralSpecificConfig, ModelKind, TokenSource,
MistralLoader, MistralSpecificConfig, MixtralLoader, MixtralSpecificConfig, ModelKind,
TokenSource,
};
pub use request::Request;
pub use response::ChatCompletionResponse;
Expand Down
Loading

0 comments on commit 00ac1e1

Please sign in to comment.