diff options
Diffstat (limited to 'candle-transformers/src/models/mistral.rs')
| -rw-r--r-- | candle-transformers/src/models/mistral.rs | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/candle-transformers/src/models/mistral.rs b/candle-transformers/src/models/mistral.rs index e8f7a7c4..f927f88b 100644 --- a/candle-transformers/src/models/mistral.rs +++ b/candle-transformers/src/models/mistral.rs @@ -1,3 +1,10 @@ +//! Mixtral Model, based on the Mistral architecture +//! +//! See Mistral and Mixtral at: +//! - [Hugging Face](https://huggingface.co/docs/transformers/model_doc/mixtral) +//! - [Github](https://github.com/mistralai/mistral-src) +//! + use crate::models::with_tracing::{linear_no_bias, Linear, RmsNorm}; /// Mistral LLM, https://github.com/mistralai/mistral-src use candle::{DType, Device, Module, Result, Tensor, D}; |